HDOJ 1308.Is It A Tree?

2015-07-15
问题简述：

　　给出一组节点关系，判断由这些节点组成的图是否为一颗树。

　　树只有一个根节点，每个节点只有一条边指向它，没有环。

　　原题链接：http://poj.org/problem?id=1308

解题思路：

　　使用并查集判断是否只有一个根节点是很简单的——让并查集种祖先的父亲是他自己即可方便计算其数量，一旦祖先数量超过一，它就不是树；

　　也可使用并查集判断图是否有环——当两个即将要链接的节点都有相同的祖先时，这就产生了一个环。

源代码：

/*
OJ: POJ
ID: forever
TASK: 1308.Is It A Tree?
LANG: C++
NOTE: Find-Union
*/
#include <cstdio>

const int MAX=10005;
int father[MAX],sign[MAX],flag;

int Find(int x) {
if(father[x]==x)
return x;
else
return Find(father[x]);
}

void Union(int x,int y) {
x=Find(x);
y=Find(y);
if(x!=y)
father[x]=y;
else
flag=0;
}

int main()
{
int a,b,k=1;
while(scanf("%d %d",&a,&b)) {
if(a==-1&&b==-1) break;
if(a==0&&b==0) {
printf("Case %d is a tree.\n",k++);
continue;
}

flag=1;
int m=0;
for(int i=0;i<MAX;i++) {
father[i]=i;
sign[i]=0;
}
Union(a,b);
sign[a]=sign[b]=1;

while(scanf("%d %d",&a,&b)) {
if(a==0&&b==0) break;
if(a>m)
m=a;
if(b>m)
m=b;
Union(a,b);
sign[a]=sign[b]=1;
}
int sum=0;
for(int i=1;i<MAX;i++) {
if(sign[i]&&father[i]==i)
sum++;
if(sum>1) {
flag=0;
break;
}
}
if(flag)
printf("Case %d is a tree.\n",k++);
else
printf("Case %d is not a tree.\n",k++);
}
return 0;
}