小希的迷宫(并查集)

并查集

小希的迷宫

• 题目

  小希想要设计一个迷宫，保证任意两个房间有且仅有一条路径相通。先给出设计图，判断是否符合要求

• 输入

  3 8  6 8  6 4
  5 3  5 6  5 2  0 0
  

• 输出

  No
  

由于任意两个房间之间仅有一条路径相通，所以一定不存在环，即所有的点都在一个集合之中。对于输入的两点，如果$find(a)==find(b)$，则说明已经在同一集合中，再连接就会构成回路。

• 注意：任意两个房间相通还意味着所有的点都应在一个集合中，所以最终还需要判断有几个集合。

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 10;
int fa[maxn], vis[maxn];//用vis记录点是否出现过
bool flag;
void init()
{
	for (int i = 1; i < maxn; i++) {
		fa[i] = i;
		vis[i] = 0;
	}
	flag = true;
}
int find(int x)
{
	return x == fa[x] ? fa[x] : find(fa[x]);
}
void union_set(int x, int y)
{
	x = find(x);
	y = find(y);
	if (x != y) {
		fa[x] = y;
	} else {
		flag = false;
	}
}
void check()
{
	int cnt = 0;
	for (int i = 1; i < maxn; i++) {
		if (vis[i] && fa[i] == i) {
			cnt++;
		}
		if (cnt > 1) {
			flag = false;
			break;
		}
	}
}
int main()
{
	ios::sync_with_stdio(0), cin.tie(0);
	int x, y;
	init();
	while (cin >> x >> y)
	{
		if (x == -1 && y == -1) {
			break;
		}
		if (x == 0 && y == 0) {
			check();
			if (flag)	cout << "Yes" << endl;
			else	cout << "No" << endl;
			init();
		}
		else {
			vis[x] = vis[y] = 1;
			union_set(x, y);
		}
	}
}