首页 > 代码库 > 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小希的迷宫 【并查集】