上次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
题意:给定一棵树的对应关系,判断是不是一棵树。
思路:可以用并查集的思想,判断每条边的两个端点是否在一个集合内。如果在一个集合内:说明已经形成了回路。最后判断是不是只有一个根。 Kruskal 算法中也是用并查集判断树的。还有这道题有个坑点,如果是一个空树的话,直接输出YES。
代码如下:
#include<cstdio>
#include<cstring>
#include<map>
#include<iostream>
using namespace std;
int f[100010];
int getf(int v) //找根
{
if(f[v]==v)
return v;
return f[v]=getf(f[v]);
}
int main()
{
int a,b;
while(~scanf("%d%d",&a,&b)&&(a!=-1||b!=-1)) //格式化输入
{
if(a==0&&b==0) //空树
{
printf("Yes\n");
continue;
}
for(int i=1;i<=100010;i++) //并查集初始化
f[i]=i;
map<int,int>v; //用map记录出现过哪些端点
v[a]=v[b]=1;
f[b]=a; //把上面那对处理一下
int flag=0;
while(scanf("%d%d",&a,&b)) //格式化输入
{
if(a==0&&b==0)
break;
v[a]=v[b]=1;
int t1=getf(a);
int t2=getf(b);
if(t1!=t2) //不在同一个集合内
f[t2]=t1;
else //在同一个集合内,说明着两个端点在以前已经出现在树中了
flag=1;
}
if(flag)
printf("No\n");
else
{
map<int,int>::iterator it;
for(it=v.begin();it!=v.end();it++) //遍历出现过的顶点,看看是不是只有一个根
{
if(f[it->first]==(it->first))
flag++;
}
if(flag==1)
printf("Yes\n");
else
printf("No\n");
}
}
}