poj1308 Is It A Tree?

并查集的基础题。在加入边前检查两点之前是否是同一个集合，若是则不是一棵树。输入完后检查是否所有点属于同一个集合。

#include<cstdio>
#include<string.h>
#include<iostream>
using namespace std;
const int maxn=1<<15;
bool flag,vis[maxn];
int p[maxn],rank[maxn],data[maxn],m;
int a,b,ra,rb;

int find(int x)
{
    if(x!=p[x])
    {
        p[x]=find(p[x]);
    }
    return p[x];
}

void unions(int x,int y)
{
    if(rank[x]>rank[y])
    {
        p[y]=x;
    }
    else
    {
        p[x]=y;
        if(rank[x]==rank[y])
        rank[y]++;
    }
}
int main()
{
    //freopen("test.txt","r",stdin);
    int t=0;
    while(1)
    {
        flag=true;
        m=0;
        int i;
        memset(vis,0,sizeof(vis));
        for(i=1;i<=maxn;i++)
        {
            p[i]=i;
            rank[i]=0;
        }
        while(scanf("%d%d",&a,&b)!=EOF)
        {
            if(a==-1&&b==-1) return 0;
            if(a==0&&b==0)
            {
                int key=find(data[0]);
                for(i=1;i<m;i++)
                {
                    if(key!=find(data[i]))
                    {
                        flag=false;break;
                    }
                }
                if(flag)
                printf("Case %d is a tree.\n",++t);
                else printf("Case %d is not a tree.\n",++t);
                break;
            }
            if(!vis[a]) {data[m++]=a;vis[a]=true;}
            if(!vis[b]) {data[m++]=b;vis[b]=true;}
            if(!flag) continue;
            ra=find(a);
            rb=find(b);
            if(ra==rb) {flag=false;continue;}
            unions(ra,rb);
        }
    }
    return 0;
}