bzoj3674 可持久化并查集加强版 （主席树 & 并查集）

bzoj3674 可持久化并查集加强版

原题地址：http://www.lydsy.com/JudgeOnline/problem.php?id=3674

题意：
n个集合 m个操作
操作：
1 a b 合并a,b所在集合
2 k 回到第k次操作之后的状态(查询算作操作)
3 a b 询问a,b是否属于同一集合，是则输出1否则输出0
请注意本题采用强制在线,所给的a,b,k均经过加密,加密方法为x = x xor lastans,lastans的初始值为0

数据范围
0 < n,m<=2*10^5

题解：
维护并查集，实际上就是维护一个树形结构，要记录每个节点的fa。
回到之前的某次操作，就需要把fa数组还原。
而合并操作只会改变一个点的fa，很容易想到主席树来维护。
而不停跳fa，为保证单次getfa的复杂度，我们需要保证树高为log，就需要启发式合并。
为了启发式合并，我们还要维护一个deep。
于是就是维护fa，deep两个数组的主席树。

代码：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=100005;
const int M=10000006;
struct node
{
    int ls,rs,dep,fa;
}tr[M];
int n,m,root[M],tail=0;
void build(int &nd,int lf,int rg)
{
    nd=++tail;
    if(lf==rg) {tr[nd].dep=0; tr[nd].fa=lf; return;}
    int mid=(lf+rg)>>1;
    build(tr[nd].ls,lf,mid);
    build(tr[nd].rs,mid+1,rg);
}
int query(int nd,int lf,int rg,int pos)
{
    if(lf==rg) return nd;
    int mid=(lf+rg)>>1;
    if(pos<=mid) return query(tr[nd].ls,lf,mid,pos);
    else return query(tr[nd].rs,mid+1,rg,pos);
}
void modify(int pre,int &now,int lf,int rg,int pos,int val)
{
    now=++tail; tr[now]=tr[pre];    
    if(lf==rg) {tr[now].fa=val; return;}
    int mid=(lf+rg)>>1;
    if(pos<=mid) modify(tr[pre].ls,tr[now].ls,lf,mid,pos,val);
    else modify(tr[pre].rs,tr[now].rs,mid+1,rg,pos,val);
}
void add(int pre,int &now,int lf,int rg,int pos,int val)
{
    now=++tail; tr[now]=tr[pre];    
    if(lf==rg) {tr[now].dep+=val; return;}
    int mid=(lf+rg)>>1;
    if(pos<=mid) add(tr[pre].ls,tr[now].ls,lf,mid,pos,val);
    else add(tr[pre].rs,tr[now].rs,mid+1,rg,pos,val);
}
int getfa(int k,int x)
{
    int fx=query(root[k],1,n,x);
    if(tr[fx].fa==x) return fx;
    else return getfa(k,tr[fx].fa);
}
void merge(int k,int x,int y)
{
    int fx=getfa(k,x);
    int fy=getfa(k,y);
    if(tr[fx].fa==tr[fy].fa) return;
    int depx=tr[fx].dep; int depy=tr[fy].dep;
    if(tr[fx].dep>tr[fy].dep) swap(fx,fy);
    modify(root[k-1],root[k],1,n,tr[fx].fa,tr[fy].fa);
    if(depx==depy) add(root[k],root[k],1,n,tr[fy].fa,1);
}
int main()
{
    scanf("%d%d",&n,&m);
    build(root[0],1,n);
    int lastans=0;
    for(int i=1;i<=m;i++)
    {
        root[i]=root[i-1]; 
        int opt,x,y; scanf("%d",&opt);
        if(opt==1)
        {
            scanf("%d%d",&x,&y);x^=lastans; y^=lastans;
            merge(i,x,y);
        }
        else if(opt==2)
        {
            int pre; scanf("%d",&pre); pre^=lastans;
            root[i]=root[pre];
        }
        else
        {
            scanf("%d%d",&x,&y); x^=lastans; y^=lastans;
            int fx=getfa(i,x); int fy=getfa(i,y);
            if(fx==fy) lastans=1; else lastans=0;
            printf("%d\n",lastans);
        }
    }
    return 0;
}