小希的迷宮

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

思路：

使用並查集或者DFS判斷圖中是否有環形成，在函式union_set 裡面一旦兩個節點父節點一致即形成了環，輸出No。滿足以下要求輸出Yes：

1.無環

2.根節點就一個

初代版本：（這說明迴圈陣列比迭代器快多了）

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

const int maxn = 1e5+10;

int pre[maxn];
int vis[maxn];
bool flag;
int maxnum;
void init() {
    for (int i=0; i<maxn; i++) pre[i] = i;
    memset(vis,0,sizeof vis);
    flag = true;
}
int find(int x) {
    
 
if (x == pre[x]) return  x;
    return pre[x] = find(pre[x]);
}

void merge(int a,int b) {
    int ra = find(a);
    int rb = find(b);
    if (ra == rb) 
        flag = false;//形成環
    else 
        pre[ra] = rb;
}

int main() {
    init();
    int u,v;
    set<int>st;
    while ( scanf("%d%d",&u,&v) ) {
        if ( u == -1 && v == -1 ) return 0;

        if ( u == 0 && v == 0 ) {
            int cnt = 0;
            for (set<int>::iterator it = st.begin(); it != st.end(); it++) {
                if (vis[*it] && pre[*it] == *it) cnt++;
            }
            // for (int i=1; i<=maxnum; i++) 
            //     if (vis[i] && pre[i] == i) cnt++;
            if (flag && cnt<=1)
                cout<<"Yes"<<endl;
            else 
                cout<<"No"<<endl;
            init();
            st.clear();
            continue;
        }
        st.insert(u);st.insert(v);
        vis[u] = 1;
        vis[v] = 1;
        if (u != v) merge(u,v);
    }
    return 0;
}

更新版本：想著每次初始化得迴圈1e+5次，於是使用了vector初始化，結果速度還慢了。

#include <iostream>
#include <cstring>
#include <set>
#include <vector>
using namespace std;
const int maxn = 1e5+10;
int pre[maxn];
bool flag;
int find(int x) {
    if (x == pre[x]) return  x;
    return pre[x] = find(pre[x]);
}
void merge(int a,int b) {
    int ra = find(a);
    int rb = find(b);
    if (ra == rb) 
        flag = false;//形成環
    else 
        pre[ra] = rb;
}

int main() {
    flag = true;
    int u,v;
    set<int>st;
    vector<int> vec;
    while ( scanf("%d%d",&u,&v) ) {
        //測試資料結束
        if ( u == -1 && v == -1 ) return 0;
        if ( u == 0 && v == 0 ) {
            //初始化pre陣列
            for (set<int>::iterator it = st.begin(); it != st.end(); it++) {
                pre[*it] =*it;
            }
            //合併
            for (vector<int>::iterator vit=vec.begin()  ; vit !=vec.end(); vit++){
                int u1 = *vit;
                vit++;
                int v1 =*vit;
                if (u1 != v1) merge(u1,v1);
            }
            int cnt = 0;
            //判斷聯通圖的個數
            for (set<int>::iterator it = st.begin(); it != st.end(); it++) {
                if (pre[*it] == *it) cnt++;
            }
            if (flag && cnt<=1)
                cout<<"Yes"<<endl;
            else 
                cout<<"No"<<endl;
            flag = true;
            st.clear();
            vec.clear();
            continue;
        }
        vec.push_back(u);
        vec.push_back(v);
        st.insert(u);
        st.insert(v);
    }
}