洛谷 #2024. 食物链

题意

三个种群,A -> B -> C -> A，食物链是环

有n个动物，给出u, v同种 或 u吃v，问其中假话的个数

题解

f[x]同种群，f[x + n]天敌，f[x + 2n]食物

调试记录

若u吃v，那么u不能是v的食物

#include <cstdio>
#define maxn 50005

using namespace std;

int f[maxn * 3], n, m; //+n 天敌 +2n 食物 
int getf(int x){return x == f[x] ? x : f[x] = getf(f[x]);} 

int main(){
	scanf("%d%d", &n, &m);
	
	for (int i = 1; i <= 3 * n; i++) f[i] = i;
	
	int opt, u, v, ans = 0;
	while (m--){
		scanf("%d%d%d", &opt, &u, &v);
		
		if (u > n || v > n){
			ans++; continue;
		}
		
		if (opt == 1){
			if (getf(u) == getf(v + n) || getf(u) == getf(v + 2 * n)){
				ans++;
				continue;
			}
			f[getf(u)] = getf(v);
			f[getf(u + n)] = getf(v + n);
			f[getf(u + 2 * n)] = getf(v + 2 * n);
		}
		
		if (opt == 2){
			if (u == v){
				ans++;
				continue;
			}
			if (getf(u) == getf(v) || getf(u) == getf(v + 2 * n)){
				ans++;
				continue;
			}
			f[getf(u)] = getf(v + n);
			f[getf(u + n)] = getf(v + 2 * n);
			f[getf(u + 2 * n)] = getf(v);
		}
	}
	printf("%d\n", ans);
	
	return 0;
}