poj 2377

kruskal 最大生成树

代码如下

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
struct Node
{
    int u,v,cost;
}a[20007];
bool cmp(Node a,Node b)
{
    return a.cost>b.cost;
}
int f[1007];
int Find(int x)
{
    if(x==f[x]) return x;
    else return f[x]=Find(f[x]);
}
void unite(int x,int y)
{
    int fx=Find(x);
    int fy=Find(y);
    if(fx!=fy)
        f[fx]=fy;
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=0;i<m;i++){
        scanf("%d%d%d",&a[i].u,&a[i].v,&a[i].cost);
    }
    sort(a,a+m,cmp);
    for(int i=1;i<=n;i++)
        f[i]=i;
    int res=0;
    for(int i=0;i<m;i++){
        Node b=a[i];
        if(Find(b.u)!=Find(b.v)) {
            unite(b.u,b.v);
//            printf("%d %d\n",b.u,b.v);
            res+=b.cost;
        }
    }
    int t=0;
    for(int i=1;i<=n;i++){
        if(f[i]!=i) t++;
    }
    if(t!=n-1) {
        printf("-1\n");
    }
    else{
        printf("%d\n",res);
    }
}