HDU1223-还是畅通工程-prim算法(模板题)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1233

好吧，终于是中文题了，哈哈。。。

#include<iostream>
#include<string>
#include<cstdio>
#include<cstring>
#include<map>
#include<queue>
#include<cmath>
#include<stack>
#include<set>
#include<vector>
#include<algorithm>
#define LL long long
#define inf 1<<29
#define s(a) scanf("%d",&a)
#define CL(a,b) memset(a,b,sizeof(a))
using namespace std;
const int N=205;
int n,m,a,b,c;
string s1,s2;
int Map[N][N];
int dist[N];
bool vis[N];
int prim()
{
    CL(vis,false);
    int i,res=0;
    for(int i=1;i<=n;i++) dist[i]=Map[1][i];
    vis[1]=true;
    for(i=2;i<=n;i++){
        int v=-1;
        for(int u=1;u<=n;u++)
            if(!vis[u]&&(v==-1||dist[u]<dist[v])) v=u;
        if(v==-1) break;
        res+=dist[v];
        vis[v]=true;
        for(int u=1;u<=n;u++)
            dist[u]=min(dist[u],Map[v][u]);
    }
    if(i<=n) return -1;
    return res;
}
int main()
{
    while(~s(n)){
        if(!n) break;
        for(int i=1;i<=(n*(n-1)/2);i++){
            scanf("%d%d%d",&a,&b,&c);
            Map[a][b]=Map[b][a]=c;
        }
        printf("%d\n",prim());
    }
    return 0;
}