Hdu 3367 Pseudoforest（并查集）

标签：

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

思路：类似最大生成树，先将边按照从大到小排序。合并时，若两点不连通：若x和y都在环中，则合并后环数大于1，不合题意。若x或y在环中，则合并，并将x和y同时标记在环中。若两点连通：x或y在环中合并后环数均大于一，不合题意。若都不在，则合并，构成一个新环，并将x和y同时标记为在环中。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define debu
using  namespace std;
const int maxn=1e4+50;
const int maxm=1e5+50;
struct Node
{
    int u,v,w;
};
int n,m,ans;
int fa[maxn];
int cir[maxn];
Node e[maxm];
int cmp(Node a,Node b)
{
    return a.w>b.w;
}
int Find(int x)
{
    if(x!=fa[x]) fa[x]=Find(fa[x]);
    return fa[x];
}
void init()
{
    ans=0;
    memset(cir,0,sizeof(cir));
    for(int i=0; i<=n; i++) fa[i]=i;
}
int main()
{
#ifdef debug
    freopen("in.in","r",stdin);
#endif // debug
    while(scanf("%d%d",&n,&m)==2&&(n+m))
    {
        init();
        for(int i=0; i<m; i++)
            scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
        sort(e,e+m,cmp);
        /*for(int i=0; i<m; i++)
            cout<<i<<" "<<e[i].u<<" "<<e[i].v<<" "<<e[i].w<<endl;*/
        for(int i=0; i<m; i++)
        {
            int x=Find(e[i].u);
            int y=Find(e[i].v);
            if(x!=y)
            {
                if(cir[x]&&cir[y]) continue;
                ans+=e[i].w;
                fa[x]=y;
                if(cir[x]||cir[y])
                {
                    cir[x]=1;
                    cir[y]=1;
                }
            }
            else
            {
                if(cir[x]||cir[y]) continue;
                ans+=e[i].w;
                cir[x]=1,cir[y]=1;
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}

Hdu 3367 Pseudoforest（并查集）

标签：

原文地址：http://blog.csdn.net/wang2147483647/article/details/52074597