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;
}