C - 食物链 POJ - 1182

C - 食物链 POJ - 1182

大佬，%%%

加入给你一个 1， 那 1 应该放进 A、B、还是C呢，不知道了，所以开三倍的空间， 1~n 表示 A， n+1~2n 表示 B， 2n+1 ~ 3*n 表示 C

我以为路径压缩了就可以用 fa[i] 代替 find(i) 了，实际上，这是不对的，修改的过程中，可能会跳过一些点，所以想要 fa[i] = find(i), 必须把所有元素 find 一遍才能实现

AC 代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
const int maxn = 5e4+100;
int n, k;
int fa[3*maxn];

int find(int x) {
	if(x == fa[x]) return x;
	else return fa[x] = find(fa[x]);
}

void Union(int x, int y) {
	int xx = find(x), yy = find(y);
	if(xx == yy) return ;
	fa[yy] = xx;
}

void init() {
	for(int i = 1; i <= 3*n; i++) fa[i] = i;
}

int main() {
//	freopen("test.in", "r", stdin);
	scanf("%d%d", &n, &k);
	int cnt = 0;
	init();
	for(int i = 1; i <= k; i++) {
		int op, u, v;
		scanf("%d%d%d", &op, &u, &v);
		if(u > n || v > n || u <= 0 || v <= 0) cnt++;
		else if(op == 1){ // 因为每个元素在 A B C 中的效果是一样的，所以判断都以 A 进行即可 
			if(find(u) == find(v+n) || find(u) == find(v+2*n)) cnt++; // 如果 AB || AC 在同一集合，则错 
			else { 
				Union(u, v);
				Union(u+n, v+n);
				Union(u+2*n, v+2*n);
			} 
		}
		else if(op == 2) {
			if(find(u) == find(v) || find(u) == find(v+2*n)) cnt++; // A -> A || A -> C 
			else { // 每次合并的结果都是 A -> B 则 A 是 B 的 father 
				Union(u, v+n); // A -> B
				Union(u+n, v+2*n); // B -> C
				Union(u+2*n, v); // C -> A
			}				
		}
	}
	printf("%d\n", cnt);

	return 0;
}