HDU 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

题意： 每个编号代表一个房间，每输入两个编号代表这两个房间连通，直到输入 0 0 表示这组数据结束，输入 -1 -1 表示文件结束。要求任意两个房间必须连通，并且任意两个房间只能有一条通路。符合则输出Yes，否则输出No。
题解： 题意是非常清晰的，用并查集也非常的好做，每给出两个房间，就将其合并，归为同一类。但因为房间编号比较随机，我不知道样例会不会诡异到给你一个编号1，和一个编号100000，但是中间不连续，这样的话用数组从头遍历，其实会浪费很多不必要的时间。所有我用了set，每次输入的时候就将编号插入，结束以后遍历set里的编号即可。
！！！有个坑点在与，它可能整一组数据只有0 0，然后这种情况应该输出Yes。

#include<iostream>
#include<algorithm>
#include<set>
using namespace std;
const int N = 1e5 + 5;

int a, b, c, d, f[N];
int find(int x) { return f[x] == x ? x : f[x] = find(f[x]);}
void un(int a, int b){
    int fa = find(a), fb = find(b);
    f[fa] = fb;
}
int main(){
    for (int i = 0; i < N; i++)    f[i] = i;
    while (~scanf("%d%d", &a, &b) && a != -1){
        if(a==0 && b==0){
            printf("Yes\n");
            continue;
        }
        for (int i = 1; i < N; i++) f[i] = i;
        set<int> s;
        int flag = 0;
        un(a, b);
        s.insert(a);
        s.insert(b);
        while (~scanf("%d%d", &c, &d) && c){
            if (find(c) == find(d))
                flag = 1;
            else{
                un(c, d);
                s.insert(c);
                s.insert(d);
            }
        }
        int cnt = 0;
        for(set<int>::iterator it = s.begin(); it != s.end(); it++)
            if(find(*it) == *it){
                cnt++;
                if(cnt>1){
                    flag = 1;
                    break;
                }
            }
        if(flag)    printf("No\n");
        else        printf("Yes\n");
    }
    return 0;
}