HDOJ-小希的迷宫 水并查集

1.判断所有点的根节点是不是同一个
2.加点时判断两个点之前的祖宗是不是一个

#include<cstdio>
int am[100005],fa[100005],n,a[100005],b[100005];
bool sym;
void init()
{
    for(int i=0;i<100005;i++)
    {
     am[i] = 1;
     fa[i] = i;
    }
}
int toGetFarther(int x)
{
    while(fa[x]!=x)
        x = fa[x];
    return x;
}
void unonFind(int x,int y)
{
    int fax = toGetFarther(x);
    int fay = toGetFarther(y);
    if(fax==fay){
        sym = false;
        return ;
    }
    if(am[fax]<am[fay])
    {
        fa[fax] = fay;
        am[fay]+=am[fax];
    }
    else
    {
        fa[fay] = fax;
        am[fax]+=am[fay];
    }
}
int main()
{
    int i1 = 1;
    while(1)
    {
       init();
       sym = true;
       int i=0;
       while(scanf("%d%d",&a[i],&b[i]))
       {
         if(a[i]==-1&&b[i]==-1)return 0;
         if(a[i]==0&&b[i]==0)break;
         if(sym)unonFind(a[i],b[i]);
         i++;
       }
       if(sym)
       {
       int ans = toGetFarther(a[0]);
       int n = i;
       for(int i=1;i<n;i++)
        {
            if(!(toGetFarther(a[i])==ans))sym = false;
        }
        }
       if(sym)printf("Case %d is a tree.\n",i1);
       else printf("Case %d is not a tree.\n",i1);
       i1++;
    }
    return 0;
}