小希的迷宮 簡單並查集

小希的迷宮

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 56504    Accepted Submission(s): 17735

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

 

Author

Gardon

 

Source

HDU 2006-4 Programming Contest

題目比較簡單，只需要在判斷時候利用並查集查詢是否會有環路即可。（即如果兩個頂點不同的話，其根借點不能相同）

注意最終輸出的條件

Yes: 當直接輸入0 0時或者只有一個指向自己的頂點的時候，或者是其餘題設的情況

No: 當出現環路的情況

#include <iostream>
#include <set>
#include <cstring>
#include <cstdio>
using namespace std;

const int MAXN = 100000 + 5;
int parent[MAXN];
int tables[MAXN];
bool ok = true;
bool first = true;
int maxnum = 0;
int minnum = 100000 + 5;
set<int> nodes;

void initSet(){
    nodes.clear();
    ok = true;
    first = true;
    maxnum = 0;
    minnum = 100000 + 5;
    memset(tables,0,sizeof(tables));
    for(int i = 0; i < MAXN; i ++){
        parent[i] = i;
    }
}

int find(int x){

    if(x != parent[x]){
        int t = find(parent[x]);
        parent[x] = t;
        return t;
    }else
        return x;
}

bool unionSet(int a,int b){
    if(a == b) return true;
    int x = find(a);
    int y = find(b);

    if(x == y) return false;

    parent[x] = y;
    return true;
}



int main(){
    int u,v;
    initSet();
    while(scanf("%d %d",&u,&v) && u != -1){
         if(u == 0){
            if(first){
                cout << "Yes" <<endl;
            }else{
                int ans = 0;
                for(int i = minnum; i <= maxnum; i ++)
                    tables[find(i)] = 1;

                for(int i = minnum; i <= maxnum; i ++)
                    if(tables[i] && nodes.count(i)){
                            ans ++;
                            //cout << "root :"<<i<<endl;
                    }

                if(ok && ans == 1)
                    cout <<"Yes"<<endl;
                else
                    cout << "No"<<endl;
            }
            initSet();
         }else{
            maxnum = max(maxnum,u);
            maxnum = max(maxnum,v);
            minnum = min(minnum,u);
            minnum = min(minnum,v);
            nodes.insert(u);
            nodes.insert(v);
            if(first)
                first = false;

            if(ok)
                ok = unionSet(u,v);

         }

    }

    return 0;
}