HDU 1272 小希的迷宫 (并查集)

题目链接：请戳这里。  

解题思路：题目意思判断是不是连通无环的图，首先想到的就是并查集。

              1）判断成环的时候，只要判断输入边的两个点。若有一个共同的父节点，那么这两个点就成环。

              2）判断连通的时候，只要判断根集合的元素数是否等于总的节点数。

             PS：当输入的这组数据只有 0 0 时，依然是满足条件的，即应输出 "Yes"。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
#define N 100000+10
using namespace std;

int num[N],s;
int f[N],v[N]; 
void Init(int n)
{
    for(int i=1;i<=n;i++)
    {
        f[i]=i;
        v[i]=num[i]=1;  //v[]记录状态，起始值为1
    }
}
int Find(int x)
{
    return f[x]==x?x:f[x]=Find(f[x]);
}
int Merge(int a,int b)
{
    int ra=Find(a),rb=Find(b);
    if(ra==rb) return 1;    //存在环
    else {
        f[rb]=ra;
        num[ra]+=num[rb];
        s=num[ra];      //s记录集合中的元素数。
        return 0;
    }
}
int main()
{
    int a,b;
    while(scanf("%d %d",&a,&b),a!=-1&&b!=-1)
    {
        Init(N);
        if(a==0 && b==0){
            puts("Yes");    //~.~
            continue;
        }
        int ok=Merge(a,b),n=v[a]+v[b];  //ok记录是否存在环，n记录节点总数。
        v[a]=v[b]=0;
        while(scanf("%d %d",&a,&b),a||b)
        {
            n+=(v[a]+v[b]);
            v[a]=v[b]=0;
            ok+=Merge(a,b);
        }
        if(ok) puts("No");
        else {
            if(s==n) puts("Yes");
            else puts("No");
        }
    }
    return 0;
}