HDOJ1272(并查集，判断是否为树)

0 0 Yes

1 1 0 0 Yes

1 2 2 1 0 0 No

//自回路不算一条边的！ 居然有 0 0 这样的测试数据

#include<iostream>
#include<cstdio>
#include<cstring>
#include<set>
using namespace std;
set<int> ans;
set<int> member;
const int SIZE=+;
int par[SIZE];
int rnk[SIZE];
void init(int n)
{
    for(int i=; i<=n; i++)
    {
        par[i]=i;
        rnk[i]=;
    }
}

int fnd(int x)
{
    if(par[x]==x)
        return x;
    return par[x]=fnd(par[x]);
}

void unite(int x, int y)
{
    int a=fnd(x);
    int b=fnd(y);

    if(a==b)
        return ;
    if(rnk[a]<rnk[b])
    {
        par[a]=b;
    }
    else
    {
        par[b]=a;
        if(rnk[a]==rnk[b])
            rnk[a]++;
    }
}

int main()
{

    int x,y;
    int cnt=;
    init(SIZE-);
    int flag=;
    while(scanf("%d %d",&x, &y)!=EOF&&x!=-&&y!=-)
    {
        if(x==&&y==)
        {
            if(flag==)
            {
                printf("Yes\n");
                continue;
            }

            if(member.size()!=cnt+)
            {
                printf("No\n");
            }
            else
            {

                for(set<int>:: iterator it=member.begin(); it!=member.end(); it++)
                {
                    ans.insert(fnd(*it));
                }
                if(ans.size()==)
                    printf("Yes\n");
                else
                    printf("No\n");
            }

            member.clear();
            ans.clear();
            cnt=;
            init(SIZE-);
            flag=;
        }
        else
        {
            flag=;
            unite(x,y);
            member.insert(x);
            member.insert(y);
            if(x!=y)
            {
                cnt++;
            }
        }

    }

    return ;
}