并查集 4104 这是一棵树吗

判断是不是树

输入：

每输入一对都为0的数时，表示一组数据输入完毕。每条边有一对正整数表示，第一个数为有向边的起始边，第二个数为有向边的终止点。一对负数的输入就表示输入的结束。

输出：

每组测试数据输出一行判断结果，若输入的图为树，则输出“Case k is a tree.”，否则输出“Case k is not a tree.”其中k表示每组数据的编号（编号从1开始）。

样例输入：

6 8 5 3 5 2 6 4

5 6 0 0

8 1 7 3 6 2 8 9 7 5

7 4 7 8 7 6 0 0

3 8 6 8 6 4

5 3 5 6 5 2 0 0

-1 -1

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
//int cmp(const void*a,const void*b)
//{
//    return(*(int *)a)<(*(int *)b)?1:-1;
//}
int f[1000000];
int flag[1000000];
int find(int x)
{
    if(f[x]!=x)
    f[x]=find(f[x]);
    return f[x];
}

int join(int x,int y)
{
    int a=find(x);
    int b=find(y);
    if(b!=y)
    return 1;
    if(a==b)
    return 1;
    else
    f[b]=a;
    return 0;
}

int main()
{
    int a,b;
    int t=0,key=0,max=0,g=0,p=1;
    memset(flag,0,sizeof(flag));
    for(int i=1;i<=999999;i++)
    f[i]=i;
    while(1)
    {
        scanf("%d%d",&a,&b);
        if(a>max)
        max=a;
        if(b>max)
        max=b;
        if(a<=-1&&b<=-1)
        break;
        if(a==0&&b==0)
        {
            for(int i=1;i<=max;i++)
            {
                if(flag[i]==1&&f[i]==i)
                g++;
            }
            if(key>0||g>=2)
            printf("Case %d is not a tree.
",p++);
            else
            printf("Case %d is a tree.
",p++);
            t=0;key=0;max=0;g=0;
            memset(flag,0,sizeof(flag));
            for(int i=1;i<=999999;i++)
            f[i]=i;
        }
        else
        {
            flag[a]=1;
            flag[b]=1;
            key+=join(a,b);
        }
    }

    return 0;
}