数据结构查漏补缺

1.并查集

1.带权并查集：例题

解法：d数组是节点到根的距离，算出d数组然后相减

void init(){
	for(int i=1;i<=30001;i++){
		fa[i]=i;
		sz[i]=1;
	} 
}
int get(int x){
	if(x==fa[x]) return x;
    int f=get(fa[x]);
    d[x]+=d[fa[x]];//d原本是x~fa[x]的距离，算出fa[x]到根的距离后来更新d数组
	return fa[x]=f;
}
void add(int x,int y){
	int k1=get(x);
	int k2=get(y);
	if(k1!=k2){
		fa[k1]=k2;
		d[k1]=sz[k2];//先让d是x~fa[x]的距离
		sz[k2]+=sz[k1];
	}
}

2.种类并查集 

原理：根据不同的限制条件将并查集中的一个点分成多个点

例题

#include<bits/stdc++.h>
using namespace std;
int fa[150011],a,b,c,d,m,n,ans;//将一个点分为三个，分别是自己，敌人和食物
void init();
int get(int x);
void add(int x,int y);
bool ok(int x,int y);
bool ok2(int x,int y);
int main(){
	scanf("%d%d",&a,&b);
	init();
	for(int i=1;i<=b;i++){
		int o,k,w;
		scanf("%d%d%d",&w,&o,&k);
		if(o>a||k>a){
			ans++;
			continue;
		}
		if(w==1){
			if(ok(o,k)){//是一个种族，合并自己，敌人，食物
				add(o,k);
				add(o+a,k+a);
				add(o+a*2,k+a*2);
			}else{
				ans++;
			}
		}
		if(w==2){
			if(ok2(o,k)){//不是一个种族，a是b的敌人，b是a的食物，a的敌人（c）是b的食物
				add(o+a,k);
				add(k+a*2,o);
				add(k+a,o+a*2);
			}else{
				ans++;
			}
		}
	}
	printf("%d",ans);
	return 0;
}
void init(){
	for(int i=1;i<=a*3+5;i++) fa[i]=i;
} 
int get(int x){
	if(x==fa[x]) return x;
	return fa[x]=get(fa[x]);
}
void add(int x,int y){
	x=get(x);
	y=get(y);
	if(x!=y) fa[x]=y;
}
bool ok(int x,int y){
	int ex=get(x+a);
	int ey=get(y+a);
	int ox=get(x+a*2);
	int oy=get(y+a*2);
	if(ex==get(y)||ey==get(x)) return false;//同一个种族不能互相吃
	return true;
}
bool ok2(int x,int y){
	int ex=get(x+a);
	int ey=get(y+a);
	int ox=get(x+a*2);
	int oy=get(y+a*2);
	if(get(x)==get(y)||ey==get(x)) return false;//a吃b，ab不能一个种族且b不能吃a
	return true;
}