hdu 3367 伪森林（kruskal)

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

求pseudoforest伪森林，要求每个连通分量最多可以有一个环。求能构成的最大值

我是用kruskal的方法按照求最大生成树那样求的，只不过要加一个判断，就是判断两颗子树是够成环，如果各成环，就不能合并，如果只有其中一个成环或者都不成环，那么就可以合并，并对其进行标记。。。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
const int N=10010;
using namespace std;

int n,m;
int parent[N];
int visited[N];//用来标记是否存在环

struct Edge {
    int u,v,w;
    bool operator < (const Edge &p) const {
        return p.w<w;//从大到小
    }
}edge[N*10];


//初始化
void UFset(){
    for(int i=0;i<n;i++)
        parent[i]=-1;
}

int Find(int x){
    int s;
    for(s=x;parent[s]>=0;s=parent[s]);
    //路径压缩，优化
    while(s!=x){
        int temp=parent[x];
        parent[x]=s;
        x=temp;
    }
    return s;
}

//合并
void Union(int R1,int R2){
    int r1=Find(R1);
    int r2=Find(R2);
    int temp=parent[r1]+parent[r2];
    if(parent[r1]>parent[r2]){
        parent[r1]=r2;
        parent[r2]=temp;
    }else {
        parent[r2]=r1;
        parent[r1]=temp;
    }
}


int main(){
    while(scanf("%d%d",&n,&m)!=EOF){
        if(n==0&&m==0)break;
        memset(visited,0,sizeof(visited));
        int u,v,ans=0;
        for(int i=0;i<m;i++){
            scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
        }
        sort(edge,edge+m);
        UFset();
        for(int i=0;i<m;i++){
            u=Find(edge[i].u);
            v=Find(edge[i].v);
            if(u!=v){
                if(visited[u]&&visited[v])continue;  //都存在环
                if(visited[u]||visited[v])
                    visited[u]=visited[v]=1;
                ans+=edge[i].w;
                Union(u,v);
            }else if(!visited[u]||!visited[v]){
                Union(u,v);
                visited[u]=visited[v]=1;
                ans+=edge[i].w;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}