上次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
并查集模板题
看过同学的代码才发现是并查集模板题,又了解了一个知识点。
注意如果第一组数据输入的n和m都等于0,也要输出“Yes”。
用一维数组F[ ]标记各个房间的根结点。
在输入过程中,求出最大的房间编号maxx,用并查集求出最小树,在输入过程中,标记各个房间号(因为可能有的房间编号没有被输入过)。使用并查集,判断两个房间A和B的根结点是否相同,若不相同,更新B的根结点F[B]=F[A],使它们的根结点相同,若输入的两个房间号的根结点相同,说明 如果加上这条边会形成回路,P=1。
若p=1,输出"No",continue。
从1到maxx遍历标记过的结点,若出现不止一个根结点,说明房间没有连通,输出“No”,否则输出“Yes”。
#include<string.h>
#include<stdio.h>
#include<algorithm>
using namespace std;
int f[100010],p,book[100010];
struct note
{
int x,y;
} que[100010];
int getf(int v)
{
if(f[v]==v)
return v;
f[v]=getf(f[v]);
return f[v];
}
int merge(int a,int b)
{
int t1=getf(a);
int t2=getf(b);
book[a]=book[b]=1; //一定要标记
if(t1!=t2)
{
f[t2]=t1;
return 1;
}
return 0;
}
int main()
{
int n,m;
int maxx=-1;
while(~scanf("%d%d",&n,&m) && n!=-1 && m!=-1)
{
memset(book,0,sizeof(book));
for(int i=1; i<100000; i++)
f[i]=i;
if(n==0&&m==0)
{
printf("Yes\n");
continue;
}
maxx=max(maxx,max(n,m));
merge(n,m);
p=0;
while(~scanf("%d%d",&n,&m)&&(n+m))
{
maxx=max(maxx,max(n,m));
if(!merge(n,m)) //如果n点和m点的根结点相同,证明若加上a和b这条边则会形成回路
p=1;
}
if(p)
{
printf("No\n");
continue;
}
int sum=0;
for(int i=1; i<=maxx; i++)
{
if(book[i] && f[i]==i) //如果被标记过,并且该点是根结点,sum++
sum++;
}
if(sum==1)
printf("Yes\n"); //如果只出现一个根结点,证明房间已经连通,并且未出现回路
else
printf("No\n");
}
return 0;
}