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

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。
HDU-1272 小希的迷宫 ( 并查集 )
 
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对迷宫的根节点进行标记,方便进行迷宫的合并和统计操作。
 
 1 #include<iostream>
 2 #include<cstring>
 3 using namespace std;
 4 
 5 int father[100010];
 6 int a, b;
 7 bool ff[100010];
 8 
 9 int findfather( int x ){
10     while( x != father[x] )
11         x = father[x];
12     return x;
13 } 
14 
15 int main(){
16     ios::sync_with_stdio( false );
17     
18     while( cin>> a >> b, a != -1 && b != -1 ){
19         int flag = 1;
20         int ans = 0;
21         
22         if(a == 0 && b == 0) cout<<"Yes"<<endl;
23         else{
24             memset( ff, 0, sizeof(ff) );
25         
26             for( int i = 0; i < 100010; i++ )
27                 father[i] = i;
28             
29             father[a] = b;    
30             ff[b] = 1;
31             ans++;
32             
33             while( cin >> a >> b, a !=0 && b != 0 ){
34                 if( !flag ) continue;
35                 
36                 a = findfather( a );
37                 b = findfather( b );
38                 if( a == b ) 
39                     flag = 0;
40                 else {
41                     father[a] = b;
42                     if( ff[a] && ff[b] ) ans--;
43                     else if( ff[a] || ff[b] ) ans += 0;
44                     else ans++;
45                     ff[a] = 0;
46                     ff[b] = 1; //将每个迷宫用其根节点标记,然后合并 
47                 }
48             }
49             
50             if( flag && ans == 1 ) cout << "Yes
";
51             else cout << "No
";
52         }
53     }
54     
55     return 0;
56 }