hdoj1325 Is It A Tree?

Is It A Tree?题目链接

题意：
多组测试数据， 每组数据有多个数对， 表示一条有向边（即第一个数是第二个数的父节点）， 以 0,0 为一组测试数据结束标志。当输入-1，-1时测试结束。 从那些给出的信息中判断是否是一棵树。
分析：
1、只可以有一个根节点， 也可以是一个点都没有的空树；
2、除了根节点， 每个点只有一个父节点。
3、因为只可以有一个父节点， 所以我们可以把一个合法的关系对（一个父亲节点与孩子节点对）合并到一个集合里， 他们的父节点都标记为所属树的根节点， 那样我们方便判断是否成环。
4、判断是否成环， 成环不可以。
5、 还有一个很衰的地方， 注意不光输入-1，-1 测试结束， 只要输入两个数都小于0就结束。

#include<iostream>
#include<cstdio>
#include<string.h>
#include<math.h>
#include<algorithm>
using namespace std;

const int N = 10005;

int a, b, n, mi, mx, sum, flag, v
, pre
;
int find(int x)//寻找x所属的树的根节点
{
int r, i, j;
r = x; i = x;
while(r != pre[r])
r = pre[r];
while(pre[i] != r)
{
j = pre[i];
pre[i] = r;
i = j;
}
return pre[x];
}
int main()
{
n = 0;
while(scanf("%d%d", &a, &b) != EOF)
{
memset(v, 0, sizeof(v));//记录所涉及的点
for(int i = 1; i <= N; i++)
pre[i] = i;
if(a < 0 && b < 0) break;
else if(a == 0 && b == 0)//特殊处理， 空树
{
printf("Case %d is a tree.\n", ++n);
continue;
}
pre[b] = a;
v[a] = 1;
v[b] = 1;
mi = min(a, b);
mx = max(a, b);
flag = 1;
while(scanf("%d%d", &a, &b) != EOF)
{
if(a == 0 && b == 0) break;
if(mx < a || mx < b) mx = max(a, b);
if(mi > a || mi > b) mi = min(a, b);
v[a] = 1;
v[b] = 1;
if(flag == 1)
{
int fx = find(a);
int fy = find(b);
if((fy != b) || (fy == fx))//若构成环， 或孩子节点已经有父节点了， 那就是不合法的边
flag = 0;
else if(fy == b && fy != fx)//如果不构成环并且没有父亲节点， 那就是合法的， 加入
pre[b] = fx;
}
}
int sum = 0;
if(flag == 1)
{
for(int i = mi; i <= mx; i++)//判断是否存在多棵树
{
if(v[i] == 1)
{
int fx = find(i);
if(fx == i)
{
sum++;
if(sum > 1)
{
flag = 0;
break;
}
}
}
}
}
if(flag == 1)
printf("Case %d is a tree.\n", ++n);
else if(flag == 0)
printf("Case %d is not a tree.\n", ++n);
}
return 0;
}

View Code