BZOJ3673 可持久化并查集 by zky 【主席树】

BZOJ3673 可持久化并查集 by zky

---

Description

n个集合 m个操作
操作：
1 a b 合并a,b所在集合
2 k 回到第k次操作之后的状态(查询算作操作)
3 a b 询问a,b是否属于同一集合，是则输出1否则输出0
0<n,m<=2∗104$0<n,m<=2\ast {10}^4$

Sample Input

5 6
1 1 2
3 1 2
2 0
3 1 2
2 1
3 1 2

Sample Output

1
0
1

---

就是并查集吗，只需要可持久化一个数组就完了

然后主席树可以很方便地实现

模拟一下并查集操作模式就好了

---

#include<bits/stdc++.h>
using namespace std;
#define N 200010
int n,m,tot=0;
int rt[N],ls[N],rs[N],val[N],dep[N];
void build(int &t,int l,int r){
    if(l>r)return;
    t=++tot;
    if(l==r){val[t]=l;dep[t]=1;return;}
    int mid=(l+r)>>1;
    build(ls[t],l,mid);
    build(rs[t],mid+1,r);
}
void modify(int &t,int last,int l,int r,int pos,int vl){
    t=++tot;
    if(l==r){val[t]=vl;dep[t]=dep[last];return;}
    ls[t]=ls[last];
    rs[t]=rs[last];
    int mid=(l+r)>>1;
    if(pos<=mid)modify(ls[t],ls[last],l,mid,pos,vl);
    else modify(rs[t],rs[last],mid+1,r,pos,vl);
}
int query(int t,int l,int r,int pos){
    if(l==r)return t;
    int mid=(l+r)>>1;
    if(pos<=mid)return query(ls[t],l,mid,pos);
    else return query(rs[t],mid+1,r,pos);
}
void add(int t,int l,int r,int pos){
    if(l==r){dep[t]++;return;}
    int mid=(l+r)>>1;
    if(pos<=mid)add(ls[t],l,mid,pos);
    else add(rs[t],mid+1,r,pos);
}
int find(int t,int x){
    int p=query(t,1,n,x);
    if(x==val[p])return p;
    return find(t,val[p]);
}
int main(){
    scanf("%d%d",&n,&m);
    build(rt[0],1,n);
    for(int i=1;i<=m;i++){
        int op,x,y;scanf("%d",&op);
        if(op==1){
            rt[i]=rt[i-1];
            scanf("%d%d",&x,&y);
            int fax=find(rt[i],x);
            int fay=find(rt[i],y);
            if(val[fax]==val[fay])continue;
            if(dep[fax]>dep[fay])swap(fax,fay);
            modify(rt[i],rt[i-1],1,n,val[fax],val[fay]);
            if(dep[fax]==dep[fay])add(rt[i],1,n,val[fay]);
        }else if(op==2){
            scanf("%d",&x);
            rt[i]=rt[x];
        }else if(op==3){
            rt[i]=rt[i-1];
            scanf("%d%d",&x,&y);
            int fax=find(rt[i],x);
            int fay=find(rt[i],y);
            if(val[fax]==val[fay])printf("1\n");
            else printf("0\n");
        }
    }
    return 0;
}