kuangbin题单 并查集 种类并查集 食物链 POJ1182

再次做经典题
day1 打卡

方法一 带权并查集

刚写完上一题，很自然的就想到用带权并查集解决问题，设x与y为给定的两种动物，可以想到有两种关系：一是x与y之前已经确定过关系，二是x与y之前没有确定过关系。
如果是第一种情况，我们根据权值确定之前记录的关系是否满足现在给定的关系，如果不满足假话数加一。
如果是第二种情况，我们可以将x与y进行合并，确定一个为另一个的父亲节点，同时更新w[x],w[y]中的权值。我们可以想象合并后的x与y都有一个共同的根（祖先）节点，计算x与y到根节点的差值对三取模，即为x与y的关系，总共有三种情况，我们可以规定0为同级，1为x吃y，2为y吃x。
（我自己有一步逻辑上的错误，WA了好几发，就是有两个条件，一是给定的值大于n，二是在谁吃谁的时候x与y的值相等，一开始我把这两个用if语句单独列出来判断，导致最后的假话数量被多加了几次，后来才反应过来，tcl）

AC代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=50010;
int f[maxn];
int w[maxn];
int ans=0;
int find(int x){
	if(x==f[x]) return x;
	int fx=f[x];
	f[x]=find(f[x]);
	w[x]=w[x]+w[fx];
	return f[x];
}
void add(int x,int y,int val){
	int xx=find(x);
	int yy=find(y);
	if(xx!=yy) f[xx]=yy;
	w[xx]=val+w[y]-w[x];
}
int main(){
	int n,k;
	scanf("%d%d",&n,&k);
	for(int i=0;i<=n;i++) f[i]=i;
	while(k--){
		int opp,x,y;
		scanf("%d%d%d",&opp,&x,&y);
		if(x>n||y>n) ans++;
		else{
			int xx=find(x);
			int yy=find(y);
			if(opp==1){
				if(xx!=yy){
					add(x,y,3);
				}else{
					if((w[x]-w[y])%3!=0) ans++;
				}
			}else{
				if(x==y) ans++;
				else{
					if(xx!=yy){
						add(x,y,1);
					}else{
						if(abs(w[x]-w[y]-1)%3!=0) ans++;
					}
				}
			}
		}
	}
	printf("%d",ans);
	return 0;
}

方法二 种类并查集

搜到的大佬讲解，一下就明白了种类并查集为什么要开关系倍的数组：种类并查集讲解
种类并查集即是有多少层关系，就用多少倍的数组存这些关系，如本题一种动物和其它动物可以有三种关系，分别是同类，捕食，天敌，我们的数组就开三倍的大小，去储存这些关系，然后判断。

AC代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=50010;
int f[3*maxn];
int ans=0;
int find(int x){
	while(x!=f[x]) x=f[x];
	return x;
}
int main(){
	int n,k;
	scanf("%d%d",&n,&k);
	for(int i=0;i<=3*n;i++) f[i]=i;
	while(k--){
		int opp,x,y;
		scanf("%d%d%d",&opp,&x,&y);
		if(x>n||y>n){
			 ans++;
			 continue;
		} 
		if(opp==1){
			if(find(x+n)==find(y)||find(x+2*n)==find(y)){
				ans++;
				continue;
			}
			f[find(x)]=find(y);
			f[find(x+n)]=find(y+n);
			f[find(x+2*n)]=find(y+2*n);
		}else if(opp==2){
			if(find(x)==find(y)||find(x+2*n)==find(y)){
				ans++;
				continue;
			}
			f[find(x)]=find(y+2*n);
			f[find(x+n)]=find(y);
			f[find(x+2*n)]=find(y+n);
		}
	}
	printf("%d",ans);
	return 0;
}