HDU 1325，POJ 1308 Is It A Tree

//Is It A Tree?
#include<stdio.h>
#include<cmath>
#include<string.h>
#include<iostream>
#include<algorithm>
#define INF 0x3f3f3f3f
using namespace std;
#define maxn 1005
int pre[maxn],visB[maxn],root,vis[maxn],MAX,flag=0;
void iint()
{
    for(int i=0; i<=maxn; i++)
    {
        pre[i]=i;
        visB[i]=0;
        vis[i]=0;
    }
    MAX=0;  flag=0;
}
int find(int x)
{
    return x==pre[x]?x:find(pre[x]);
}
void join(int x,int y)
{
    int fx=find(x),fy=find(y);
    if(fx!=fy)  pre[fx]=fy;
}
int main()
{
    int a,b,Case=1/*,real_flag1=0*/,FLAG=0;
    iint();
    while(~scanf("%d%d",&a,&b),a>=0&&b>=0)
    {
        if(a==0&&b==0&&flag==0)
        {   //空树也是树
            printf("Case %d is a tree.\n",Case++);
            iint();
            continue;
        }
        flag=1;
        visB[b]++;
        vis[a]=vis[b]=1;

        MAX=max(MAX,a);    MAX=max(MAX,b);
        /*for(int i=1; i<=MAX; i++)
        {
            if(vis[i]&&visB[i]>1)  real_flag1=1;
            //判断根节点是否都有且只有一次，或者说父节点只有一个
            if(vis[i]&&visB[i]==0)   real_flag2++;
            //非环时，最高父节点不会被指向，所以不被指的有且只有1个
            //POJ会WA可能是，如3->1  2->1都算一个树了，2333
        }*/
        /*if(real_flag1)   FLAG=1;*/   //天真

        if(find(a)==find(b)&&a!=0&&b!=0) FLAG=1;
       记住：1->2 ,3->2时候也是树，所以判断是否为环，直接在每次读       入两个数字时候看它们的最高父节点是否相等，相等泽成环，比如
1->2,2->3,3>1.
         
        join(a,b);

        if(a==0&&b==0)
        {
            int ans=0;
            for(int i=1; i<=MAX; i++)
                if(vis[i]&&pre[i]==i)  ans++;

            if(ans!=1||FLAG)  printf("Case %d is not a tree.\n",Case++);
            else  printf("Case %d is a tree.\n",Case++);

            iint();
            /*real_flag1=0;*/ FLAG=0;
            continue;
        }
    }
    return 0;
}