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[N], pre[N];
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);
            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);
        else if(flag == 0)
            printf("Case %d is not a tree.
", ++n);
    }
    return 0;
}