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

BZOJ

---

SOL

可持久化并查集？难道可持久化非旋treap维护并查集森林？其实没那么麻烦。。可以看做是主席树的应用。

从代码的角度思考并查集，我们只需要可持久化 $fa$数组即可。
这里不进行路径压缩（版本太多，感觉压缩了实测效果也不是那么理想)
所以要按秩合并，再用主席树维护一个$dep$数组即可。

---

CODE

/**************************************************************
    Problem: 3674
    User: Jun
    Language: C++
    Result: Accepted
    Time:972 ms
    Memory:158324 kb
****************************************************************/
 
#include<bits/stdc++.h>
using namespace std;
#define sf scanf
#define pf printf
#define ll long long
#define cs const
#define db double
#define ri register int
#define gc getchar()
#define in red()
inline int red(){
    int num=0,f=1;char c=gc;
    for(;!isdigit(c);c=gc)if(c=='-')f=-1;
    for(;isdigit(c);c=gc)num=(num<<1)+(num<<3)+(c^48);
    return num*f;
}
cs int N=8e6+10,M=2e5+10;
#define lc(x) ch[x][0]
#define rc(x) ch[x][1]
 
int ch[N][2],fa[N],dep[N],tot,rev[N],rt[M],n,m;
inline int cpy(int x){++tot;ch[tot][0]=ch[x][0];ch[tot][1]=ch[x][1];dep[tot]=dep[x];return tot;}
inline void upt(int &p,int pre,int l,int r,int k,int v){
    p=cpy(pre);
    if(l==r)return fa[p]=v,rev[p]=k,void();
    int mid=(l+r)>>1;
    if(k<=mid)upt(lc(p),lc(pre),l,mid,k,v);
    else upt(rc(p),rc(pre),mid+1,r,k,v);
}
inline void add(int p,int l,int r,int k){
    if(l==r)return ++dep[p],void();
    int mid=(l+r)>>1;
    if(k<=mid)add(lc(p),l,mid,k);
    else add(rc(p),mid+1,r,k);
}
inline int query(int p,int l,int r,int k){
    if(l==r)return p;
    int mid=(l+r)>>1;
    if(k<=mid)return query(lc(p),l,mid,k);
    else return query(rc(p),mid+1,r,k);
}
inline void build(int &p,int l,int r){
    p=++tot;
    if(l==r)return dep[p]=1,fa[p]=l,rev[p]=l,void();
    int mid=(l+r)>>1;
    build(lc(p),l,mid);build(rc(p),mid+1,r);
}
inline int find(int t,int x){
    int f=query(rt[t],1,n,x);
    while(fa[f]^x){
        x=fa[f];
        f=query(rt[t],1,n,x);
    }
    return f;
}
inline bool check(int t,int x,int y){
    int fx=rev[find(t,x)],fy=rev[find(t,y)];
    return fx==fy;
}
inline void merge(int pre,int x,int y){
    int fx=find(pre,x),fy=find(pre,y);
    if(rev[fx]==rev[fy])return;
    if(dep[fx]>dep[fy])swap(fx,fy);
    upt(rt[pre+1],rt[pre],1,n,rev[fx],rev[fy]);
    if(dep[fx]==dep[fy])add(rt[pre+1],1,n,rev[fy]);
}
 
signed main(){
    n=in;m=in;
    build(rt[0],1,n);
    for(ri i=1;i<=m;++i){
        int op=in,a=in,b;
        if(op==1)b=in,merge(i-1,a,b);
        if(op==2)rt[i]=rt[a];
        if(op==3)rt[i]=rt[i-1],b=in,cout<<check(i,a,b)<<'\n';
    }
 
 
 
    return 0;
}