【并查集】hdu 1325 Is It A Tree?

注意以下4种情况:

0 0              可以
1 1 0 0       不可以
1 2 1 2 0 0 不可以
1 2 2 1 0 0 不可以

#include <cstdio>
#include <cstring>

const int MAXN=100000+5;

int a,b,mycase=1;
bool ok=true;
int father[MAXN];//记录父节点
int temp[MAXN];//判断是否为森林（用并查集）
bool vis[MAXN];

void init(){
    ok=true;
    for(int i=0;i<MAXN;i++){
        father[i]=i;
        temp[i]=i;
        vis[i]=0;
    }
}
int main()
{
    init();
    while( scanf("%d %d",&a,&b)!=EOF && a>-1 && b>-1 ){

        if(ok==false && a && b)continue;

        if(a==0 && b==0){
            int root=0;//根节点的个数
            for(int i=0;i<MAXN;i++){
                if(vis[i] && temp[i]==i)
                    root++;
            }
            if(root>1)ok=false;

            if(ok)printf("Case %d is a tree.
",mycase++ );
            else printf("Case %d is not a tree.
",mycase++ );
            
            init();
            continue;
        }

        if(a!=b && father[b]==b && father[a]!=b){
            father[b]=a;
            temp[b]=a;
            vis[a]=vis[b]=true;
        }
        else ok=false;

    }

    return 0;
}