1. 程式人生 > >Is It A Tree?(並查集)

Is It A Tree?(並查集)

題目:

  給出一對對的數字a,b,表示從a到b有一條邊。判斷這是不是一棵樹。

  • 多case,每個case以0 0 結尾
  • 輸入以-1 -1結尾

分析:①每個節點(除了根結點)只有一個入度;②只有一個根結點。

森林:多個根。

入度:指向同一個結點的邊數。

將點逐個加入集合中,然後判斷就行了。

#include<stdio.h>
#include<algorithm>
using namespace std;
int flagt;//判斷是否成樹
int vis[10005];//標記用過的結點
int pre[10005];
int find(int x)//壓縮路徑 並查詢
{
    return pre[x]==x?x:pre[x]=find(pre[x]);
}
void Merge(int x,int y)
{
    int fx=find(x);
    int fy=find(y);
    if(fy!=fx)
        pre[fy]=fx;
}
void init()
{
    int i;
    for(i=1;i<=10005;i++)
    {
        pre[i]=i;
        vis[i]=0;
    }
}
int judge_tree(int n) //判斷是否只有一個根結點。
{
    int i=1;
    while(i<=n && !vis[i])
        i++;

    int f1=find(i);

    while(i<=n)
    {
        if(vis[i] && find(i)!=f1)
            return 0;
        i++;
    }

    return 1;
}
int main()
{
    int i,j,k=1;
    int x,y,range;
    flagt=1;
    range=0;
    init();

    while(scanf("%d%d",&x,&y)!=EOF)
    {


        if(x<0 || y<0)
            return 0;

        if(x==0 || y==0) //輸入0 0時判斷。
        {
            if(flagt && judge_tree(range))
                printf("Case %d is a tree.\n",k++);

            else
                printf("Case %d is not a tree.\n",k++);
            flagt=1;
            range=0;
            init();
            continue;
        }

        if(!flagt)
            continue;

       range=max(range,max(x,y));//所涉及到的最數值最大的點。

        vis[x]=vis[y]=1;
        
        if(find(x)==find(y))//他們有共同的根結點,並且他們之間還有父子關係(即有邊相連),則無法形成樹。
            flagt=0;

        Merge(x,y);

    }
    return 0;
}