[判断是否有环,是否有多个根节点] [kuangbin带你飞]专题五 并查集 M - 小希的迷宫

M - 小希的迷宫

上次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 <cstdio>
#include <cstring>
using namespace std;
const int maxn=1e5;
int pre[maxn+10],vis[maxn+10],flag,a,b,num;
void init()
{
	flag=0,a=100010,b=-1,num=0;
	for(int i=1;i<=maxn;i++)
	{
		pre[i]=i;
		vis[i]=0;
	}
}
int find(int x) // !!!这是最重要的地方,找到祖先,把所有的子孙变成祖先;
{
	if(pre[x]!=x) pre[x]=find(pre[x]);
	return pre[x];
}
void join(int x,int y)  //合并,要是父亲已经一样了,就会变成环,flag标记
{
	int fx=find(x),fy=find(y);
	if(fx!=fy) pre[fx]=fy;
	else flag=1;
}
int main()
{
	init();
	int n,m;
	while(~scanf("%d%d",&n,&m))
	{
		if(n==-1&&m==-1) break;
		a=min(a,n);  //减少不必要的遍历,得到最小最大城市值
		b=max(b,m);
		if(n==0&&m==0)
		{
			for(int i=a;i<=b;i++)
			{
				if(vis[i]&&pre[i]==i) num++; //出现过又是根节点的增加
			}
			if(flag==1||num>1) printf("No\n"); //是环或者有多个根节点就不是树
			else printf("Yes\n");
			init();   //一切的初始化
			continue;
		}
		join(n,m);   //要是不跳出循环,就一直输入,合并并标记
		vis[n]=1,vis[m]=1;
	}
}

猜你喜欢

转载自blog.csdn.net/qq_41037114/article/details/81039868