Bzoj 3673: 可持久化并查集 by zky(主席树+启发式合并)

3673: 可持久化并查集 by zky
Time Limit: 5 Sec Memory Limit: 128 MB
Description
n个集合 m个操作
操作：
1 a b 合并a,b所在集合
2 k 回到第k次操作之后的状态(查询算作操作)
3 a b 询问a,b是否属于同一集合，是则输出1否则输出0
0<=n,m<=2*10^4
Input
Output
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
HINT
Source
出题人大SB

/*可持久化线段树+启发式合并. 可持久化线段树维护当前状态下集合的关系和秩的信息.所谓的秩就是以该元素为代表元的所有元素中的最大深度.然后按秩合并的目的是为了降常. 每个叶节点维护一颗线段树 合并的时候在权值线段树的子节点加一个数,相当于连了一条边 表示有关系存在.要先查询要将合并两个元素的父亲所在位置.显然只有在两个集合秩相同时才更新秩.*/#include<iostream>#include<cstdio>#define MAXN 20001using namespace std;int n,m,tot,root[MAXN],s[MAXN];struct data{int lc,rc,deep,x;}tree[MAXN*20];int read(){    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-48,ch=getchar();    return x*f;}void build(int &k,int l,int r){    k=++tot;    if(l==r){tree[k].x=l;return ;}    int mid=(l+r)>>1;    build(tree[k].lc,l,mid);    build(tree[k].rc,mid+1,r);    return ;}int query(int k,int l,int r,int x){    if(l==r) return k;    int mid=(l+r)>>1;    if(x<=mid) return query(tree[k].lc,l,mid,x);    else return query(tree[k].rc,mid+1,r,x);}int find(int root,int x){    int p=query(root,1,n,x);    if(x==tree[p].x) return p;    else return find(root,tree[p].x);}void change(int &k,int last,int l,int r,int x,int y){    k=++tot;tree[k].lc=tree[last].lc,tree[k].rc=tree[last].rc;    if(l==r) {        tree[k].x=y;tree[k].deep=tree[last].deep;        return ;    }    int mid=(l+r)>>1;    if(x<=mid) change(tree[k].lc,tree[last].lc,l,mid,x,y);    else change(tree[k].rc,tree[last].rc,mid+1,r,x,y);    return ;}void updata(int k,int l,int r,int x){    if(l==r){tree[k].deep++;return ;}    int mid=(l+r)>>1;    if(x<=mid) updata(tree[k].lc,l,mid,x);    else updata(tree[k].rc,mid+1,r,x);    return ;}void union_s(int l1,int l2,int i){    if(tree[l1].deep>tree[l2].deep) swap(l1,l2);    change(root[i],root[i-1],1,n,tree[l1].x,tree[l2].x);    if(tree[l1].deep==tree[l2].deep) updata(root[i],1,n,tree[l2].x);    return ;}int main(){    int x,y,z,l1,l2;    n=read(),m=read();    build(root[0],1,n);    for(int i=1;i<=m;i++)    {        z=read();        if(z==1)        {            x=read(),y=read();            root[i]=root[i-1];            l1=find(root[i],x),l2=find(root[i],y);            if(tree[l1].x!=tree[l2].x) union_s(l1,l2,i);        }        else if(z==2) x=read(),root[i]=root[x];        else {            x=read(),y=read();            root[i]=root[i-1];            l1=find(root[i],x),l2=find(root[i],y);            if(l1==l2) printf("1\n");            else printf("0\n");        }    }    return 0;}