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
按照房间之间的限制描述再结合给的图片观察就可以知道,要判断建出来的迷宫(图Graph)是否满足要求,其实就是要判断这个图是不是一棵树,仅此而已。
那么什么是树呢?对于本题,只要满足这两个要求就能判定图是为一棵树:
- 图中不能存在环。
- 图只能存在一个连通块。
因此,可以借助并查集,将所有结点慢慢归并到一个集合(生成树)里,那么好处是,在建图的时候,对于两个将要建边的结点,先判断它们是否已经在同一集合里(其实就是同一树上),如果已经在了,那么注定加入这条边后构成环,说明这个图不合理,返回“No”
。
如果前面的建图都没问题,在最后图建好后,只需判断图是否就只有一个连通块(每找到一个结点等于自己祖先结点的情况,就说明找到了一个新的连通块),如果连通分量超过1,说明这个图不合理,返回“No”
。
排除掉上面两个情况后,说明建图合理,返回“Yes”
。
C++代码
#include <iostream>
#include <set>
using namespace std;
const int N = 1e5 + 10;
/**
* 坑点:图中的点的编号既可能不从1开始,也可能不连续,所以需要存一下点的编号
*/
int fa[N];
set<int> point; //存下点的编号
int a, b;
bool flag; //是否有建边的冲突
void init(){
point.clear();
for(int i = 1;i <= N;i ++) fa[i] = i; //并查集的初始化
flag = false;
}
int find(int x){
return x == fa[x] ? fa[x] : find(fa[x]);
}
void Union(int x, int y){
x = find(x), y = find(y);
if(x != y){
fa[x] = y;
}
}
int main(){
init();
while(cin >> a >> b){
if(a == -1) break;
if(!a){ //一组数据读完,输出答案并为下一组初始化
int cnt = 0; //连通块的数量
for(auto p : point){
if(p == fa[p]) cnt ++;
if(cnt > 1){
flag = true;
break;
}
}
if(flag) cout << "No" << endl;
else cout << "Yes" << endl;
init();
continue;
}
if(find(a) == find(b)) //a和b已经在一个集合里了
flag = true;
Union(a, b);
point.insert(a), point.insert(b);
}
return 0;
}