首页 > 代码库 > 1272小希的迷宫 【并查集】
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
并查集的核心代码：
int find(int a){
int r = a;
while(set[r] != r)
r = set[r];
int i=a;
int j;
while(i!=r){
j = set[i];
set[i] = r;
i = j;
}
return r;
}
并查集以数据连接为特点，代码实现是：通过数组下标和对应的值，这两个数构成一条连接。比如：
a[3] = 6;
a[6] = 4;
表示3的父节点是6，6的父节点是4，3和6构成连接，6和4构成连接
核心解释：
//先从一个低节点，扫到高节点，如果高节点还有连接就继续往上扫，
//如果没有或者说还没建立连接，就暂停。
//扫完之后下面还有一个循环，
//这个循环用来把高节点的值直接赋予它下面的所有节点。
//并不用担心先后顺序，先后顺序不同有可能改变最高节点，只要这棵树不循环，依旧只有一个最高节点。
//但如果你想把某两个节点相邻串在一起，但是本来就是一个集合（一串）的，，就会出现find(a) = find(b)的状况，就是说，他们共同拥有一个最高节点，如果a,b相连，就会形成循环。
这道题有些其他特点，就是他不一定只有一个集合，就是说，他有可能有多个最高结点，这是不允许的。所以：
for(int i=0;i<=100005;i++){
if(mark[i] && set[i] == i)
c++;
}
凡是最高结点，set[i] == i 但如果有多个就必定会c > 1。
而mark[i]是用来记录结点是否存在的，因为循环从0到100005，有可能数据中并没有某个数字的结点，但他的set[i] == i，这并不能看作是最高结点。所以两个条件配合在一起，只有满足该结点存在并认为是最高结点时，c才能加一。
如下：
#include<stdio.h>
int set[100005];
int mark[100005];
int find(int a){
int r = a;
while(set[r] != r)
r = set[r];
int i=a;
int j;
while(i!=r){
j = set[i];
set[i] = r;
i = j;
}
return r;
}
int merge(int a,int b){
int A,B;
A = find(a);
B = find(b);
if(A != B){
set[B] = A;
return 1;
}
else
return 0;
}
void initialize(){
for(int i=0;i<=100005;i++){
set[i] = i;
mark[i] = 0;
}
}
int main(){
int a,b,k = 0;
int sign = 1;
initialize();
while(~scanf("%d %d",&a,&b)){
if(a == -1 && b == -1)
break;
if(k == 0){
if(a == 0 && b == 0){
printf("Yes\n");
continue;
}
}
k++;
if(a != 0 && b != 0){
mark[a] = mark[b] = 1;
if(merge(a,b) == 0){
sign = 0;
}
}else if(a == 0 && b == 0){
if(sign == 0){
printf("No\n");
sign = 1;
}else{
int c = 0;
for(int i=0;i<=100005;i++){
if(mark[i] && set[i] == i)
c++;
}
if(c == 1)
printf("Yes\n");
else
printf("No\n");
}
k = 0;
initialize();
}
}
return 0;
}
1272小希的迷宫 【并查集】