hdu 3367 Pseudoforest

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

#include<cstdio>
#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;

const int maxn = 10005;
const int maxm = 100050;

struct Edge{
    int u,v,c;
    bool operator < (const Edge & rh) const{
        return c > rh.c;
    }
};
vector<Edge> e;
int p[maxn];
int n,m; 
bool flag[maxn];

int find(int x){
    return x == p[x] ? x : p[x] = find(p[x]);
}


int main()
{
    //if(freopen("input.txt","r",stdin)== NULL)  {printf("Error\n"); exit(0);}

    while(cin>>n>>m && (n || m)){  //这儿也有点问题啊 
        e.clear();
        for(int i=1;i<=m;i++){
            int u,v,c;
            scanf("%d%d%d",&u,&v,&c);
            e.push_back((Edge){u,v,c});
        }
        sort(e.begin(),e.end());
        for(int i=0;i<n;i++) p[i] = i;
        memset(flag,0,sizeof(flag));
        long long ans = 0; 
        for(int i=0;i<m;i++){
            int a = find(e[i].u),b = find(e[i].v); 
            //printf("%d %d %d %d \n",e[i].u,e[i].v,a,b);
            if(a != b && (!flag[a] || !flag[b])){ 
            //这儿出现问题，没考虑到a所在的树与b所在的树不在同一个树上，但是两个数都有一个环 
               if(flag[a])  p[b] = a;    //这儿出问题了；
               else         p[a] = b;
                       
               ans += e[i].c;    
            }
            else if(a == b && !flag[a]){
               ans += e[i].c;
               flag[a] = true;
            }
                  // printf("%d\n",ans);
        }
        cout<<ans<<endl;
    }
}