hdu 3367 Pseudoforest

http://acm.hdu.edu.cn/showproblem.php?pid=3367

/*
    题目大意：给出 n 个点 和m 条边，现在要求边的最大和，要求最多有一个环存在。
        环可以用并查集判断。因为要求边的权值和最大，我们可以先把所有的边按照权值由大到小排序
*/
#include <cstdio>
#include <cstring>
#include<iostream>
#include <algorithm>
using namespace std;
#define M 10010
using namespace std;
int vis[M],pre[M];
struct A
{
    int x,y,len;
    A(){}
    A(int x,int y,int len):x(x),y(y),len(len){}
    bool operator <(const A& rhs)const
    {
        return len>rhs.len;
    }
}a[10*M];//保存边

void init(int n)
{
    for(int i = 0; i < n; i++)
    {
        pre[i] = i;
        vis[i] = 0;
    }
}
int find(int x)
{
    return x != pre[x]? pre[x] = find(pre[x]):x;
}

int main()
{
    int n,m,i;
    while(cin>>n>>m)
    {
        if(n + m == 0) break;
        memset(vis,0,sizeof(vis));
        init(n);
        for(i = 0; i < m; i++)
        scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].len);
        sort(a,a+m); // 对边排序

        int ans = 0;
        for(i = 0; i < m; i++)
        {
            int x = find(a[i].x);
            int y = find(a[i].y);
            if(x == y) //如果边的左端点的祖先 和右断点 祖先一样，说明已经出现了一个环
            {
                if(vis[x]) continue;//如果下次再出现这个中的任意一条边，都不能要
                vis[x] = 1;//出现环了，就要标记祖先
            }
            else
            {
                if(vis[x] && vis[y]) continue;//如果两个端点的祖先不相同，但是他们的祖先都标记过，这条边同样不能要
                if(!vis[x]) pre[x] = y;
                else pre[y] = x;//如果左端点已经标记过，这条边我们选取了，就要把左端点归到右端点去
            }
            ans += a[i].len;
        }
        cout<<ans<<endl;
    }

    return 0;
}