EOJ2529 强大的lwc

http://acm.cs.ecnu.edu.cn/problem.php?problemid=2529

题意，求修造为通路耗费的最小精力。

即求无向图的最小生成树，此题注意结点是0到n 为n+1个，判断连通条件可以判断树的边数是不是为n，此题用prim

#include<map>
#include<cmath>
#include<cctype>
#include<cstdio>
#include<string>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int inf=999999999;
int mat[105][105];
int n;
void init(){
    for(int i=0;i<=n;i++)
        for(int j=0;j<=n;j++)
            mat[i][j]=inf;
}
int prim(){                                  //注意 ，都是从0-n
    int lowcost[105],s[105];
    int ans=0,cnt=0;
    for(int i=0;i<=n;i++){
        lowcost[i]=mat[0][i];
        s[i]=0;
    }
    s[0]=1;
    for(int i=0;i<=n;i++){
        int min=inf,k;
        for(int j=0;j<=n;j++)
            if(!s[j] && lowcost[j]<min){
                min=lowcost[j];
                k=j;
            }
        if(min==inf) break;
        ans+=min;
        cnt++;
        s[k]=1;
        for(int j=0;j<=n;j++)
            if(!s[j] && mat[k][j]<lowcost[j])
                lowcost[j]=mat[k][j];
    }
    if(cnt==n)           //判断是否连通
        return ans;
    else return 0;
}
int main(){
    int t;
    cin>>t;
    while(t--){
        int m;
        scanf("%d%d",&n,&m);
        init();
        while(m--){
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            if(c<mat[a][b])
                mat[a][b]=mat[b][a]=c; //注意是无向
        }
        /*for(int i=0;i<=n;i++){
            for(int j=0;j<=n;j++)
                printf("%d ",mat[i][j]);
            cout<<endl;
        }*/
        int ans=prim();
        if(ans)printf("%d\n",ans);
        else printf("-1\n");
    }
    return 0;
}