poj 2377 拉最长的线问题 kruskal算法

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define maxn 1111
#define maxm 22222
struct node
{
    int u,v,w;
}edge[maxm];
int T,n,m,fa[maxn];
int cmp(node a,node b)
{
    return a.w>b.w;
}
int find(int x)
{
    if(fa[x]==x) return x;
    return fa[x]=find(fa[x]);
}
int kruskal(int n,int m)
{
    for(int i=1;i<=n;i++)fa[i]=i;
    int ans=0,cnt=0;
    sort(edge,edge+m,cmp);
    for(int k=0;k<m;k++)
    {
        int x=find(edge[k].u),y=find(edge[k].v);
        if(x!=y)
        {
            cnt++;
            fa[x]=y;
            ans+=edge[k].w;
            if(cnt==n-1)return ans;
        }
    }
    return -1;
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        for(int i=0;i<m;i++)scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
        printf("%d\n",kruskal(n,m));
    }
    return 0;
}