HDOJ 1325 并查集

跟小希的迷宫基本一样，只是此题是有向图，要注意：1无环 2 只有一个入度为0的结点（根结点），

不存在入度大于1的结点。输入结束条件是两个负数，而不是-1，不然会TLE。

#include<stdio.h>
#define NUM 23
int root[NUM], visit[NUM], lu[NUM];
void init(){
    for(int i=0; i<=NUM; i++){
        root[i]=i;
        visit[i]=0;
        lu[i]=0;
    }
}
int find(int x){
    while(root[x]!=x)
        x=root[x];
    return x;
}
void merge(int a, int b){
    int x=find(a);
    int y=find(b);
    root[x]=y;
}
int main(){
    int a,b,i,flag,count;
    init();
    flag = 1;//judge
    count = 0;//numcase
    while(EOF != scanf("%d%d",&a,&b)){
        if(a <0 || b < 0)   break;
        if(visit[a] == 0)   visit[a] = 1;
        if(visit[b] == 0)   visit[b] = 1;
        lu[b]++;
        //如果是相同的根，会形成环
        if(find(a) == find(b) && a != 0 && b != 0 && a != b) flag = 0;
        else    merge(a,b);
        if(a==0 && b==0){
            count++;
            int cnt=0;
            for(i=1; i<NUM; i++){
                if(visit[i] == 1 && root[i] == i)
                    cnt++;
                if(cnt > 1 || lu[i] > 1) //判断是否有2个以上的根 && 不能有入度大于1的点
                    flag = 0;
            }
            if(!flag)
                printf("Case %d is not a tree.
",count);
            else printf("Case %d is a tree.
",count);
            init();
            flag=1;
        }
    }
    return 0;
}