191113-扩展域并查集

191113-扩展域并查集

T1 食物链

解析

因为有三种关系，同类，吃，被吃，所以我们将每个点拆成三个点:$x，xeat，xenemy$,分别表示x的同类，x的猎物，x的天敌，于是每次对于询问1，就判断xeat与y是否同族，x与yeat是否同族，如果皆不同，那么为真话，否则为假，如果为真，就将x,y合并，xeat,yeat合并，xenemy，yenemy合并。对于询问2，同理可得

该题也可以使用带权并查集做，至于解析，先咕掉
先放篇解析

题解

#include<bits/stdc++.h>
using namespace std;
int xen,xeat,yen,yeat,n,k,ans,fa[150009],d;
int read(){
	int f=1,re=0;
	char ch;
	for(ch=getchar();!isdigit(ch)&&ch!='-';ch=getchar());
	if(ch=='-'){
		f=-1;
		ch=getchar();
	}
	for(;isdigit(ch);ch=getchar())
		re=(re<<3)+(re<<1)+ch-'0';
	return re*f;
}
int getfa(int x){
	if(fa[x]==x) return x;
	return fa[x]=getfa(fa[x]);
}
void unim(int x,int y){
	int v=getfa(x);
	int u=getfa(y);
	if(u!=v) fa[u]=v;
}
int main(){
	int x,y;
	n=read(),k=read();
	for(int i=1;i<=3*n;i++) fa[i]=i;
	for(int i=1;i<=k;i++){
		d=read(),x=read(),y=read();
		xen=x+n,xeat=x+2*n;
		yen=y+n,yeat=y+2*n;
		if(d==2&&x==y){
			ans++;
			continue;
		}
		if(x>n||y>n){
			ans++;
			continue;
		}
		if(d==1){
			if(getfa(xeat)==getfa(y)||getfa(x)==getfa(yeat)){
				ans++;
				continue;
			}
			unim(x,y);
			unim(xen,yen);
			unim(xeat,yeat);
		}
		if(d==2){
			if(getfa(x)==getfa(y)||getfa(x)==getfa(yeat)){
				ans++;
				continue;
			}
			unim(xeat,y);
			unim(xen,yeat);
			unim(x,yen);
		}
	}
	printf("%d\n",ans);
	return 0;
}

T2 奇偶游戏

解析

因为每次会给一个区间的奇偶性，我们将这个区间假设为$[l,r]$，如果$[l,r]$为奇，那么$su{m}_{l-1},su{m}_r$奇偶性不同$(sum表示前缀奇偶性)$
反之，那么$su{m}_{l-1},su{m}_r$奇偶性相同，我们同样是用拆点的方法，将一个点拆成$x_{odd}$和$x_{even}$，同T1一样，判断，合并即可

当然，该题也可以用带权并查集来做，至于解析，先咕掉
先放篇解析

题解

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<iomanip>
#include<algorithm>
using namespace std;
struct qrx{
	int x,y,z;
}a[500009];
int n,m,cnt,fa[40009],b[40009];
char s[109]; 
int getfa(int x){
	if(fa[x]==x) return x;
	return fa[x]=getfa(fa[x]);
}
void unio(int x,int y){
	int u=getfa(x);
	int v=getfa(y);
	fa[u]=v;
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;i++){
		scanf("%d%d%s",&a[i].x,&a[i].y,s);
		if(s[0]=='o') a[i].z=1;
		else a[i].z=0;
		b[++cnt]=a[i].x;
		b[++cnt]=a[i].y;
	}
	sort(b+1,b+cnt+1);
	int len=unique(b+1,b+cnt+1)-b-1;
	for(int i=1;i<=len*2;i++) fa[i]=i;
	for(int i=1;i<=m;i++){
		int x=lower_bound(b+1,b+len+1,a[i].x-1)-b;
		int y=lower_bound(b+1,b+len+1,a[i].y)-b;
		int x_=x+len;
		int y_=y+len;
		if(a[i].z){
			if(getfa(x)==getfa(y)){
				printf("%d\n",i-1);
				return 0;
			}
			unio(x_,y);
			unio(x,y_);
		}
		else{
			if(getfa(x)==getfa(y_)){
				printf("%d\n",i-1);
				return 0;
			}
			unio(x,y);
			unio(x_,y_);
		}
	}
	printf("%d",m);
	return 0;
}

T3 银河英雄传说（带权并查集）

解析

再合并的同时，再记录每个集合的size，以及集合中每个点到根的距离，即dep，然后对于每次询问，如果在同一列中，那么ans就等于两者dep差的绝对值

题解

#include<bits/stdc++.h>
using namespace std;
int fa[30009],size[30009],dep[30009],n,ans;
char s[10];
int read(){
	int f=1,re=0;
	char ch;
	for(ch=getchar();!isdigit(ch)&&ch!='-';ch=getchar());
	if(ch=='-'){
		f=-1;
		ch=getchar();
	} 
	for(;isdigit(ch);ch=getchar())
		re=(re<<3)+(re<<1)+ch-'0';
	return re*f;
}
int getfa(int x){
	if(fa[x]==x) return x;
	int f=getfa(fa[x]);
	dep[x]+=dep[fa[x]];
	return fa[x]=f;
}
void merge(int x,int y){
	int u=getfa(x);
	int v=getfa(y);
	fa[u]=v;
	dep[u]=size[v];
	size[v]+=size[u];
	size[u]=0;
}
int main(){
	int x,y;
	n=read();
	for(int i=1;i<=30000;i++) fa[i]=i,size[i]=1;
	for(int i=1;i<=n;i++){
		scanf("%s",s);
		x=read(),y=read();
		if(s[0]=='M') merge(x,y);
		else{
			int u=getfa(x);
			int v=getfa(y);
			if(u!=v){
				printf("-1\n");
				continue;
			}
			printf("%d\n",abs(dep[x]-dep[y])-1);
		}
	}
	return 0;
}