1. 程式人生 > >【並查集判環】HDU1272小希的迷宮【判環模板】

【並查集判環】HDU1272小希的迷宮【判環模板】

題目連結:http://acm.hdu.edu.cn/showproblem.php?pid=1272

Problem Description 上次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

程式碼:

#include<iostream>
using namespace std;
int pre[100005];
int vis[100005];
int Find(int a)     //  這裡不進行壓縮;
{
    int r=a;
    while(pre[r]!=r) r=pre[r];
    return r;
}
void mix(int x,int y)
{
    pre[x]=y;
}
int main()
{
    int a,b;
    cin.sync_with_stdio(false);
    while(cin>>a>>b&&(a!=-1&&b!=-1)){
        if(a==0&&b==0){         //  這裡是需要注意的,如果直接就為0,0直接輸出Yes;
            cout<<"Yes"<<endl;
            continue;
        }
        for(int i=1;i<100005;i++)
            pre[i]=i,vis[i]=0;
        int x=Find(a);
        int y=Find(b);
        vis[a]=vis[b]=1;
        if(x!=y) mix(x,y);
        int flag=1;
        while(cin>>a>>b&&(a&&b)){
            x=Find(a);
            y=Find(b);
            if(x==y) flag=0;        //  如果找到父節點相同的點,說明存在環;
            else mix(x,y);
            vis[a]=vis[b]=1;
        }
        if(!flag){
            cout<<"No"<<endl;
        }else{
            for(int i=1;i<100001;i++){      //  父節點只有一個;
                if(vis[i]&&pre[i]==i) flag++;
                if(flag>2) break;
            }
            if(flag==2) cout<<"Yes"<<endl;
            else cout<<"No"<<endl;
        }
    }
    return 0;
}