BZOJ 3673——可持久化并查集by zky

BZOJ 3673——可持久化并查集by zky

不同于普通并查集的是

可持久化并查集是在主席树上实现的，

原理与普通并查集一样。

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
#define maxn 20050
int cnt=0,ls[maxn*20],rs[maxn*20],val[maxn*20],root[maxn],sz,key[maxn*20],n,m;
int read(){
    int w=0;char c=getchar();
    while(c<48||c>57) c=getchar();
    while(c>=48&&c<=57){
        w=w*10+c-48;
        c=getchar();
    }
    return w;
}
void build(int &p,int l,int r){
    p=++cnt;
    if(l==r) {key[p]=l;return;}
    if(l!=r){
        int mid=(l+r)/2;
        build(ls[p],l,mid);
        build(rs[p],mid+1,r);
    }
}
void Modify(int &p,int cmp,int l,int r,int loc,int val){
    p=++cnt;ls[p]=ls[cmp];rs[p]=rs[cmp];
    if(l==r){
        key[p]=val;
    }
    if(l!=r){
        int mid=(l+r)>>1;
        if(loc<=mid){
            Modify(ls[p],ls[cmp],l,mid,loc,val);
        }
        else{
            Modify(rs[p],rs[cmp],mid+1,r,loc,val);
        }
    }
}
inline int  query(int version,int l,int r,int loc){
    int mid;
    while(l!=r){
        mid=(l+r)>>1;
        if(loc<=mid) version=ls[version],r=mid;
        else version=rs[version],l=mid+1;
    }
    return key[version];
}
int find(int version,int loc){
    int fa;
    while((fa=query(version,1,n,loc))!=loc){
        loc=fa;
    }
    return loc;
}
int main()
{
    int x,y,l,r,k,i,opt;
    n=read();m=read();
    build(root[0],1,n);
    for(i=1;i<=m;i++){
        opt=read();x=read();
        if(opt==1){
            y=read();
            Modify(root[i],root[i-1],1,n,find(root[i-1],x),find(root[i-1],y));
        }
        else if(opt==2) root[i]=root[x];
        else{
            y=read();
            root[i]=root[i-1];
            if(find(root[i],x)==find(root[i],y)) printf("1\n");
            else printf("0\n");
        }
    }
}