利用並查集判斷一個有向圖是否成樹

hdu 1325

此題與hdu 1272類似。

但此題需注意以下幾點：

1.當輸入兩個負數時退出，不是當輸出-1 -1 （這點不注意將一直超時T_T）

2.對樹的判斷除了不成環，還需注意一個問題：此圖為有向圖：當b的根不是他自己或a時，b 這個節點有兩個跟，同樣不成數。

    if(f1 == f2 || f2 != b)
        flag = 1;

 

#include<stdio.h>
#include<string.h>
#define N 110000
int f[N],vis[N];
int flag = 0;
int Maxx = 0;
int num = 1;
int Max(int a,int b)
{
    return a>b?a:b;
}
void star1()
{
    int i;
    for(i = 0;i < N;i ++)
    {
        f[i] = i;
    }
    memset(vis,0,sizeof(vis));
    flag = 0;
    Maxx = 0;
}
int root(int i)
{
    if(f[i] == i)
        return i;
    else
        return root(f[i]);
}
void U(int a,int b)
{
    int f1,f2;
    f1 = root(a);
    f2 = root(b);
    if(f1!= f2)
        f[f2] = f1;
    if(f1 == f2 || f2 != b)
        flag = 1;
}
int main()
{
    int n,m,i,j,k;
    int a,b;
    star1();
    while(~scanf("%d %d",&a,&b)&& a>=0 && b>=0)
    {
        if(a==0 && b ==0)
        {
            k = 0;
            for(i = 1;i <= Maxx;i ++)
            {
                if(vis[i] && f[i] == i)
                    k ++;
            }
            if(k <= 1 && flag == 0)
            {
                printf("Case %d is a tree.\n",num++);
            }
            else
            {
                printf("Case %d is not a tree.\n",num++);
            }
            star1();
        }
        else
        {
            Maxx = Max(Maxx,Max(a,b));
            vis[a] = vis[b] = 1;
            U(a,b);
        }
    }
    return 0;
}