HDUOJ-1325 Is It A Tree?（并查集）

Problem Description

Input

Output

Sample Input

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

Sample Output

Case 1 is a tree.
Case 2 is a tree.
Case 3 is not a tree.

输入x,y表示有向边x->y，要求无环，只有一个根，所有节点连通（根能到达所有节点）

多组输入，每组以0 0结束，所有输入以-1 -1结束

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;
const int INF = 0x3f3f3f3f;
const int N = 1e3+10;
int pre[N];
bool vis[N]= {0};//表示节点是否被指向过
int findd(int x)//找父节点
{
    if(x!=pre[x])
    {
        int t=findd(pre[x]);
        pre[x]=t;
    }
    return pre[x];
}
void unity(int x,int y)
{
    int fx = findd(x), fy = findd(y);
    if (fx != fy)
    {
        pre[fy] = fx;
    }
}
int judge(int x,int y)//判断x,y是否连通(父节点相同)
{
    int fx=findd(x),fy=findd(y);
    if(fx!=fy)return 1;
    return 0;
}
struct node//结构体存边
{
    int x;
    int y;
} no[N];
int main()
{
    int n,tem;
    int x,y;
    int fl=1;
    for(int i=1; i<=1005; i++)//初始化
    {
        pre[i]=i;
    }
    int k=0;
    int cot=1;//计数
    while(~scanf("%d%d",&x,&y))
    {
        if(x==-1&&y==-1)break;
        if(x==0&&y==0)
        {
          for(int i=1; i<k; i++)//判断是否存在不连通点，由于每次输入x,y肯定连通，判断x和上次输入y即可
            {
                if(judge(no[i-1].y,no[i].x))                
                {
                    fl=0;
                    break;
                }
            }
            if(fl)printf("Case %d is a tree.\n",cot);
            else printf("Case %d is not a tree.\n",cot);
            for(int i=1; i<=1005; i++)
            {
                pre[i]=i;
            }
            fl=1;
            k=0;
            cot++;
            for(int i=1; i<=1005; i++)vis[i]=0;
            continue;
        }
        no[k].x=x;
        no[k++].y=y;//记录每条边
        unity(x,y);
        if(vis[y])fl=0;//判断是否有唯一根和环，如果多个点指向同一点，则必然存在环或多个根节点
        vis[y]=1;
    }
    return 0;
}