1. 程式人生 > >小希的迷宮(並查集)

小希的迷宮(並查集)

題目:

上次Gardon的迷宮城堡小希玩了很久(見Problem B),現在她也想設計一個迷宮讓Gardon來走。但是她設計迷宮的思路不一樣,首先她認為所有的通道都應該是雙向連通的,就是說如果有一個通道連通了房間A和B,那麼既可以通過它從房間A走到房間B,也可以通過它從房間B走到房間A,為了提高難度,小希希望任意兩個房間有且僅有一條路徑可以相通(除非走了回頭路)。小希現在把她的設計圖給你,讓你幫忙判斷她的設計圖是否符合她的設計思路。比如下面的例子,前兩個是符合條件的,但是最後一個卻有兩種方法從5到達8。

這裡寫圖片描述

Input

輸入包含多組資料,每組資料是一個以0 0結尾的整數對列表,表示了一條通道連線的兩個房間的編號。房間的編號至少為1,且不超過100000。每兩組資料之間有一個空行。
整個檔案以兩個-1結尾。

Output

對於輸入的每一組資料,輸出僅包括一行。如果該迷宮符合小希的思路,那麼輸出”Yes”,否則輸出”No”。

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

Yes
Yes
No

題意:

其實就是判斷所給資料能不能構成一顆無向樹

這題又是用G++才A,C++ WA了。

Code:

#include<cstdio>
#include<cstdlib>
#include<iostream> #include<cstring> #include<cmath> #include<algorithm> #include<stack> #include<queue> using namespace std; const int maxn=100010; int par[maxn]; int vis[maxn]; //標記是否訪問 int ok; //標記是否成環 void init() //初始化 { for(int i=0;i<=maxn;i++) { par[i]=i; vis[i]=0
; } } int find(int x) { return x==par[x]? x:par[x]=find(par[x]); } void unite(int a,int b) { int fa=find(a); int fb=find(b); if(fa!=fb) par[fa]=fb; else ok=0; } int main() { int a,b; while(~scanf("%d%d",&a,&b)&&(a!=-1||b!=-1)) { if(!a&&!b) // 0 0 的時候也是樹 { printf("Yes\n"); continue; } init(); unite(a,b); ok=1; vis[a]=1; vis[b]=1; while(~scanf("%d%d",&a,&b),a||b) { unite(a,b); //連線a,b vis[a]=1; //標記 表示訪問過 vis[b]=1; } if(ok==0) //成環 { printf("No\n");continue; } else { int i,cnt=0; //cnt表示根節點個數 for(i=0;i<=100005;i++) { if(vis[i]&&par[i]==i) //訪問過且是根節點 cnt++; } if(cnt==1) //根節點==1 才是樹 { printf("Yes\n"); } else { printf("No\n"); } } } return 0; }