【并查集】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 不可以

 1 #include <cstdio>

 2 #include <cstring>

 3 

 4 const int MAXN=100000+5;

 5 

 6 int a,b,mycase=1;

 7 bool ok=true;

 8 int father[MAXN];//记录父节点

 9 int temp[MAXN];//判断是否为森林（用并查集）

10 bool vis[MAXN];

11 

12 void init(){

13     ok=true;

14     for(int i=0;i<MAXN;i++){

15         father[i]=i;

16         temp[i]=i;

17         vis[i]=0;

18     }

19 }

20 int main()

21 {

22     init();

23     while( scanf("%d %d",&a,&b)!=EOF && a>-1 && b>-1 ){

24 

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

26 

27         if(a==0 && b==0){

28             int root=0;//根节点的个数

29             for(int i=0;i<MAXN;i++){

30                 if(vis[i] && temp[i]==i)

31                     root++;

32             }

33             if(root>1)ok=false;

34 

35             if(ok)printf("Case %d is a tree.\n",mycase++ );

36             else printf("Case %d is not a tree.\n",mycase++ );

37             

38             init();

39             continue;

40         }

41 

42         if(a!=b && father[b]==b && father[a]!=b){

43             father[b]=a;

44             temp[b]=a;

45             vis[a]=vis[b]=true;

46         }

47         else ok=false;

48 

49     }

50 

51     return 0;

52 }