Luogu P2024 [NOI2001]食物链

#include<cstdio>
#include<cstring>
using namespace std;

int n,k,fa[1000005],ans;

inline int getfa(int v){
	return fa[v]==v?v:fa[v]=getfa(fa[v]);
}

inline void merge(int x,int y){
	x=getfa(x);y=getfa(y);
	if(x!=y)fa[y]=x;
}

inline bool family(int x,int y){
	return getfa(x)==getfa(y);
}

int main(){
	scanf("%d%d",&n,&k);
	for(int i=1;i<=3*n;i++)fa[i]=i;
	for(int i=1;i<=k;i++){
		int opt,x,y;
		scanf("%d%d%d",&opt,&x,&y);
		if(x>n||y>n){ans++;continue;}
		if(opt==1){
			if(family(x,y+n)||family(x,y+2*n)){ans++;continue;}
			merge(x,y);
			merge(x+n,y+n);
			merge(x+2*n,y+2*n);
		}
		else{
			if(x==y){ans++;continue;}
			if(family(x+2*n,y)||family(x,y)){ans++;continue;}
			merge(x+n,y);
			merge(x,y+2*n);
			merge(x+2*n,y+n);//是个环：猎物的猎物是天敌 
		}
	}
	printf("%d\n",ans);
}
//x+n:猎物 x+2*n:天敌