【主席树+并查集】P3402 可持久化并查集

这道题目需要用主席树维护可持久化，用按秩合并优化并查集即可

 

 

代码

#include<bits/stdc++.h>
using namespace std;
const int maxm=4500005;
int n,m,root[maxm],cnt,f[maxm],dep[maxm];
struct tree
{
	int l,r;
}tr[maxm];
int build(int now,int l,int r)
{
	now=++cnt;
	if(l==r)
	{
		f[now]=l;
		return now;
	}
	int mid=(l+r)>>1;
	tr[now].l=build(tr[now].l,l,mid);
	tr[now].r=build(tr[now].r,mid+1,r);
	return now;
}
void update(int &now ,int pre,int l,int r,int x,int val)
{
	now=++cnt;
	if(l==r)
	{
		f[now]=val;
		dep[now]=dep[pre];
		return;
	}
	tr[now].l=tr[pre].l;
	tr[now].r=tr[pre].r;
	int mid=(l+r)>>1;
	if(x<=mid) update(tr[now].l,tr[pre].l,l,mid,x,val);
	else update(tr[now].r,tr[pre].r,mid+1,r,x,val);
}
int query(int now,int l,int r, int x)
{
	if(l==r) return now;
	int mid=(l+r)>>1;
	if(x<=mid) return query(tr[now].l,l,mid,x);
	else return query(tr[now].r,mid+1,r,x);
}
int find(int now,int x)
{
	int father=query(now,1,n,x);
	if(x==f[father]) return father;
	return find(now,f[father]);
}
void add(int now,int l,int r,int x)
{
	if(l==r)
	{
		dep[now]++;
		return;
	}
	int mid=(l+r)>>1;
	if(x<=mid) add(tr[now].l,l,mid,x);
	else add(tr[now].r,mid+1,r,x);
}
int main()
{
    freopen("a.in","r",stdin);
    freopen("a.out","w",stdout);
    int op,x,y;
    scanf("%d%d",&n,&m);
    root[0]=build(0,1,n);
	for(int i=1;i<=m;i++)
	{
		scanf("%d",&op);
		if(op==1)
		{
			scanf("%d%d",&x,&y);
			root[i]=root[i-1];
			int f1=find(root[i],x),f2=find(root[i],y);
			if(f[f1]==f[f2]) continue;
			if(dep[f1]>dep[f2]) swap(f1,f2);
			update(root[i],root[i-1],1,n,f[f1],f[f2]);
			if(dep[f1]==dep[f2]) add(root[i],1,n,f[f2]);
		}
		if(op==2)
		{
			scanf("%d",&x);
			root[i]=root[x];
		}
		if(op==3)
		{
			root[i]=root[i-1];
			scanf("%d%d",&x,&y);
			int f1=find(root[i],x),f2=find(root[i],y);
			if(f[f1]==f[f2]) printf("1\n");
			else printf("0\n");
		}
	}
    return 0;
}