带删除并查集 学习笔记

注意：这里的删除指的是，将 \(u\) 移除集合但是其它点的关系都和移除前一样。

解法

给每个点设立一个虚点，作为他们的双亲节点。delet 的时候直接再新建一个虚点，将 \(u\) 的双亲节点设为 TA 就好了。

代码

#include<bits/stdc++.h>
#define rep(i,x,y) for(int i=x;i<=y;++i)
#define per(i,x,y) for(int i=x;i>=y;--i)
#define lon long long
using namespace std;
const int n7=301234;
int n,dcnt,T,ha[n7],vis[n7],ans;

int rd(){
	int shu=0;bool fu=0;char ch=getchar();
	while( !isdigit(ch) ){if(ch=='-')fu=1;ch=getchar();}
	while( isdigit(ch) )shu=(shu<<1)+(shu<<3)+ch-'0',ch=getchar();
	return fu?-shu:shu;
}

int geth(int z){
	while(z^ha[z])z=ha[z]=ha[ ha[z] ];
	return z;
}

void merge(int p,int q){
	int ha1=geth(p),ha2=geth(q);
	ha[ha1]=ha2;
}

void delet(int z){
	dcnt++,ha[z]=ha[dcnt]=dcnt;
}

int main(){
	n=dcnt=rd(),T=rd();
	rep(i,1,n)dcnt++,ha[i]=ha[dcnt]=dcnt;
	while(T--){
		int sys=rd(),p=rd(),q=rd();
		if(sys==1)merge(p,q);
		if(sys==2)delet(p),merge(p,q);
		if(sys==3)puts(geth(p)==geth(q)?"YES":"NO");
	}
	rep(i,1,n){
		int z=geth(i);
		if(!vis[z])vis[z]=1,ans++;
	}
	printf("%d",ans);
	return 0;
}