洛谷 P3402 【模板】可持久化并查集

题目链接：https://www.luogu.org/problem/P3402

n个集合 m个操作

操作：

• 1 a b 合并a,b所在集合

• 2 k 回到第k次操作之后的状态(查询算作操作)

• 3 a b 询问a,b是否属于同一集合，是则输出1否则输出0

#include<bits/stdc++.h>
#define en '\n'
#define ll long long
using namespace std;
const ll inf=1e18+3;
const ll maxn = 1e5+3;
const int maxm=maxn*67;
int  rd()
{
    int  x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x;
}
int tot,f[maxm],ls[maxm],rs[maxm],n,rnk[maxm],rt[maxm];
void build(int &x,int l,int r){
    x=++tot;
    if(l==r){
        f[x]=l;return ;
    }
    int mid=(l+r)>>1;
    build(ls[x],l,mid),build(rs[x],mid+1,r);
}
int ask(int x,int l,int r,int pos){
    if(l==r)return x;
    int mid=(l+r)>>1;
    if(pos<=mid)return ask(ls[x],l,mid,pos);
    return ask(rs[x],mid+1,r,pos);
}
int find(int x,int pos){
    int ind=ask(x,1,n,pos);
    if(f[ind]==pos)return f[ind];
    return find(x,f[ind]);
}
void upd(int &x,int y,int l,int r,int pos,int val){
    x=++tot;ls[x]=ls[y],rs[x]=rs[y];
    if(l==r){f[x]=val;rnk[x]=rnk[y];return ;}
    int mid=(l+r)>>1;
    if(pos<=mid)upd(ls[x],ls[y],l,mid,pos,val);
    else upd(rs[x],rs[y],mid+1,r,pos,val);
}
void add(int x,int l,int r,int pos){
    if(l==r){rnk[x]+=1;return ;}
    int mid=(l+r)>>1;
    if(pos<=mid)add(ls[x],l,mid,pos);
    else add(rs[x],mid+1,r,pos);
}
signed main()
{
    #ifdef local
    freopen("input2.txt","r",stdin);
    #endif
    n=rd();int q=rd();
    build(rt[0],1,n);
    for(int i=1;i<=q;i++){
        int op=rd();
        if(op==1){
            int x=rd(),y=rd();
            int xx=find(rt[i-1],x),yy=find(rt[i-1],y);
            if(rnk[xx]>rnk[yy])swap(xx,yy);
            upd(rt[i],rt[i-1],1,n,xx,yy);
            if(rnk[xx]==rnk[yy]){
                add(rt[i],1,n,yy);
            }
        }else if(op==2){
            int k=rd();rt[i]=rt[k];
        }
        else if(op==3){
            rt[i]=rt[i-1];
            int x=rd(),y=rd();
            int xx=find(rt[i],x),yy=find(rt[i],y);
            if(xx==yy)puts("1");
            else      puts("0");
        }
    }
    return 0;

}