杭电1272——并差集

杭电1272——并差集

原题传送门，并差集模板详解链接，大佬并差集详解链接。
解题思路
1、判断是否有环
2、判断所有的房子是否连在一起。
第一点可以通过same函数判断，在输入的时候就可以判断；
第二点可以先通过vis数组标记，然后遍历vis数组，当vis[i]==1并且id[i]==i的时候，说明i是最上面的父亲。再符合条件的情况下，这样的父亲只有一个，如果个数大于一，说明房间并不是全都相连，ans=“No”。
ac代码：

# include <iostream>
# include <algorithm>
# include <cstdio>
# include <cstdlib>
# include <string>
# include <cstring>
# include <vector>
using namespace std;

const int maxn = 1e5 + 10;
int id[maxn];
int ra[maxn];
int vis[maxn];

int fin(int x) {
	return id[x] = id[x] == x ? x : fin(id[x]);
}
void meg(int x, int y) {
	int a = fin(x);
	int b = fin(y);
	if (ra[a] < ra[b]) {
		id[a] = b;
	}
	else {
		id[b] = a;
		if (ra[a] == ra[b]) {
			ra[b]++;
		}
	}
}
bool same(int x, int y) {
	return fin(x) == fin(y);
}

int main() {
	ios::sync_with_stdio(false);
	while (true) {
		for (int i = 0; i < maxn; i++) {
			id[i] = i;
		}
		memset(ra, 0, sizeof(ra));
		memset(vis, 0, sizeof(vis));

		string ans = "Yes";
		while (true) {
			int m, n;
			cin >> m >> n;
			if (m == -1 && n == -1) {
				return 0;
			}
			if (m == 0 && n == 0) {
				break;
			}
			vis[m] = vis[n] = 1;
			if (same(m, n)) {//如果m，n已经相连，说明不符合条件
				ans = "No";
			}
			else {//如果m，n没有相连，就把他们连起来
				meg(m, n);
			}
		}
		int num = 0;
		for (int i = 0; i < maxn; i++) {//用num记录最上面的房间有几个
			if (vis[i] == 1 && id[i] == i) {
				num++;
			}
			if (num > 1) {//当num的值大于一时，说明答案已经是No，就不用继续遍历了，跳出循环。
				ans = "No";
				break;
			}
		}
		cout << ans << endl;
	}
	return 0;
}