hdu-1272 小希的迷宫（并查集）

小希的迷宫

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 83971 Accepted Submission(s): 26426

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

分析

如果输入的两个数根相同，说明已经在同一集合里面，此时再连就会造成两个房间有多条通道。题目并没有说默认在一个集合，所以如果有多个集合，则也不符合任意两个房间有且仅有一条路径可以相通的条件。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int N=100005;
int s[N];
int vis[N];
int ok;
void init()
{
	memset(vis,0,sizeof(vis));
	for(int i=0;i<N;i++){
		s[i]=i;
	}
}
int finds(int x)
{
	int r=x;
	while(r!=s[r]) r=s[r];
	int i=x,j;
	while(i!=r){
		j=s[i];
		s[i]=r;
		i=j;
	}
	return r;
}
void merge(int x,int y)
{
	int a=finds(x);
	int b=finds(y);
	if(a!=b){
		s[a]=b;
	}
	else{
		ok=0;
	}
}
int main()
{
//	freopen("data.txt","r",stdin);
	int x,y;
	while(~scanf("%d %d",&x,&y)){
		if(x==0&&y==0){
			printf("Yes\n");
			continue;
		}
		if(x==-1&&y==-1) break;
		init();
		ok=1;
		merge(x,y);
		vis[x]=1;
		vis[y]=1;
		while(scanf("%d %d",&x,&y)&&(x||y)){
			merge(x,y);
			vis[x]=1;
			vis[y]=1;
		}
		if(ok==0){
			printf("No\n");
			continue;
		}
		else{
			int cnt=0;
			for(int i=0;i<=N;i++){
				if(vis[i]&&s[i]==i)
				cnt++;
			}
			//cout<<cnt<<endl;
			if(cnt==1){
				printf("Yes\n");
			}
			else{
				printf("No\n");
			}
		}
	}
}