HDU-1272 小希的迷宫 ( 并查集 )

题目链接: 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

 

 

迷宫的要求：

不能形成回路 这点我用并查集来解决，如果加入迷宫的两个房间属于同一个集合，则不符合要求；

任意两个房间相通 虽然这一点做题的时候都知道，但是写代码的时候很容易忽视，这也是很多人WA的原因，我用数组ff对迷宫的根节点进行标记，方便进行迷宫的合并和统计操作。

 

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

int father[100010];
int a, b;
bool ff[100010];

int findfather( int x ){
    while( x != father[x] )
        x = father[x];
    return x;
} 

int main(){
    ios::sync_with_stdio( false );
    
    while( cin>> a >> b, a != -1 && b != -1 ){
        int flag = 1;
        int ans = 0;
        
        if(a == 0 && b == 0) cout<<"Yes"<<endl;
        else{
            memset( ff, 0, sizeof(ff) );
        
            for( int i = 0; i < 100010; i++ )
                father[i] = i;
            
            father[a] = b;    
            ff[b] = 1;
            ans++;
            
            while( cin >> a >> b, a !=0 && b != 0 ){
                if( !flag ) continue;
                
                a = findfather( a );
                b = findfather( b );
                if( a == b ) 
                    flag = 0;
                else {
                    father[a] = b;
                    if( ff[a] && ff[b] ) ans--;
                    else if( ff[a] || ff[b] ) ans += 0;
                    else ans++;
                    ff[a] = 0;
                    ff[b] = 1; //将每个迷宫用其根节点标记,然后合并 
                }
            }
            
            if( flag && ans == 1 ) cout << "Yes\n";
            else cout << "No\n";
        }
    }
    
    return 0;
}