codevs 1069 关押罪犯 并查集

http://codevs.cn/problem/1069/

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
struct gxx
{
    int a,b,c;
}gx[100005];
bool cmp(gxx a,gxx b)
{
    return a.c>b.c;
}
int fa[20005];
int fd[20005];
int find(int x)
{
    return fa[x]==x?x:fa[x]=find(fa[x]);
}
int main()
{
    int n,m;scanf("%d%d",&n,&m);
    int a,b,c;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d%d",&a,&b,&c);
        gx[i].a=a;
        gx[i].b=b;
        gx[i].c=c;
    }
    sort(gx+1,gx+m+1,cmp);
    for(int i=1;i<=n;i++)
        fa[i]=i;
    for(int i=1;i<=m;i++)
    {
        a=gx[i].a;
        b=gx[i].b;
        c=gx[i].c;
        if(find(a)==find(b))
        {
            printf("%d",c);
            return 0;
        }
        if(!fd[a])
            fd[a]=b;
        if(!fd[b])
            fd[b]=a;
        fa[find(a)]=find(fd[b]);
        fa[find(b)]=find(fd[a]);
    }
    printf("0");
    return 0;
}