hdu 1325/poj 1308 Is It A Tree? 并查集

题目链接

题意：给出n条有向边，判断是不是一棵树。

这题和hdu1272题目类似，只是把无向边改成了有向边。多加一个判断条件，保证子节点是根节点。

#include <iostream>
#include<cmath>
#include<cstdio>
#include<cstring>
#define N 11000

using namespace std;

int pre[N],vis[N],flag,maxn;

void init()
{
    for(int i=0;i<N;i++)    pre[i]=i;
    memset(vis,0,sizeof(vis));
    maxn=0;
    flag=1;
}

int findset(int v)
{
    int t1,t2=v;
    while(v!=pre[v])    v=pre[v];
    while(t2!=v)
    {
        t1=pre[t2];
        pre[t2]=v;
        t2=t1;
    }
    return v;
}

void unions(int u,int v)
{
    int t1=findset(u);
    int t2=findset(v);
    if(v!=t2||t1==t2)   flag=0;
    if(t1!=t2)  pre[t2]=t1;
}

int main()
{
    int kase=0;
    int u,v;
    init();
    while(~scanf("%d%d",&u,&v)&&u>=0&&v>=0)
    {
        if(u==0&&v==0)
        {
            int tp=0;
            for(int i=1;i<=maxn;i++)
                if(vis[i]&&i==pre[i]) tp++;
            if(tp<=1&&flag)
                printf("Case %d is a tree.\n",++kase);
            else
                printf("Case %d is not a tree.\n",++kase);
            init();
            continue;
        }
        maxn=max(maxn,max(u,v));
        vis[u]=vis[v]=1;
        unions(u,v);
    }
}