洛谷P2024食物链——并查集

题目：https://www.luogu.org/problemnew/show/P2024

思路：

x可以吃掉x+n，x+n可以吃掉x+2n，x+2n可以吃掉x，

【为帮助理解，可以把x看作剪刀，x+n看作布，x+2n看作石头。】

思路与关押罪犯相同（见我的博客https://mp.csdn.net/postedit/88959248）。

举例说明：

n=10，1可以吃2，则因为1与11同类，从而记11与2同类，21与12同类，1与22同类。

n=10，1与3同类，则记1与3同类，11与13同类，21与23同类。

利用并查集表达同类。

AC代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int n,m,k,x,y,fat[3*50005],ans;
/*x吃掉x+n，x+n吃掉x+2*n，x+2*n吃掉x*/ 
int father(int x){
	if( fat[x]!=x) fat[x]=father( fat[x] );
	return fat[x];
}
void unionn(int x,int y){
	int fx=father(x),fy=father(y);
	if(fx!=fy)fat[fx]=fy;
}
int main(){
	cin>>n>>m;
	for(int i=1;i<=3*n;i++)fat[i]=i;
	for(int i=1;i<=m;i++){
		scanf("%d%d%d",&k,&x,&y);
		if(x>n || y>n){
			ans++;continue;
		}
		if(k==1){
			if( father(x+n)==father(y) || 
				father(x)==father(y+n) ) ans++;
				/*x可以吃掉y，或者y可以吃掉x*/
			else{/*x与y是同类*/
				unionn(x,y);
				unionn(x+n,y+n);
				unionn(x+2*n,y+2*n);
			}	
		}
		else{/*x与y是同类，或者反过来是y吃掉x*/
			if( father(x)==father(y)||
				father(x+n)==father(y+n)||
				father(x+2*n)==father(y+2*n)||
				father(y+n)==father(x) ) ans++;
			else{/*x的食物是y，x是y的天敌，x的天敌是y的食物*/
				unionn(x+n,y);
				unionn(x,y+2*n);
				unionn(x+2*n,y+n);
			}
		}	
	}
	cout<<ans;
	return 0;
}