BZOJ3673 可持久并查集

可以用线段树维护 每个叶子结点存它父亲的位置 查一个点的父亲和正常并查集差不多 记录每个点的深度 当要相连的两个点深度相同时 （p，q） 把q的deep++ 整体用主席树维护

copy hzwer的 Orz

#include<cstdio>
#include<iostream>
using namespace std;
inline int read()
{
    int x=0;char ch=getchar();
    while(ch>'9'||ch<'0')ch=getchar();
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x;
}
int n,m,sz;
int root[200005],ls[2000005],rs[2000005],v[2000005],deep[2000005];
void build(int &k,int l,int r)
{
    if(!k)k=++sz;
    if(l==r){v[k]=l;return;}
    int mid=(l+r)>>1;
    build(ls[k],l,mid);
    build(rs[k],mid+1,r);
}
void modify(int l,int r,int x,int &y,int pos,int val)
{
    y=++sz;
    if(l==r){v[y]=val;deep[y]=deep[x];return;}
    ls[y]=ls[x];rs[y]=rs[x];
    int mid=(l+r)>>1;
    if(pos<=mid)
        modify(l,mid,ls[x],ls[y],pos,val);
    else modify(mid+1,r,rs[x],rs[y],pos,val);
}
int query(int k,int l,int r,int pos)
{
    if(l==r)return k;
    int mid=(l+r)>>1;
    if(pos<=mid)return query(ls[k],l,mid,pos);
    else return query(rs[k],mid+1,r,pos);
}
void add(int k,int l,int r,int pos)
{
    if(l==r){deep[k]++;return;}
    int mid=(l+r)>>1;
    if(pos<=mid)add(ls[k],l,mid,pos);
    else add(rs[k],mid+1,r,pos);
}
int find(int k,int x)
{
    int p=query(k,1,n,x);
    if(x==v[p])return p;
    return find(k,v[p]);
}
int main()
{
    n=read();m=read();
    build(root[0],1,n);
    int f,k,a,b;
    for(int i=1;i<=m;i++)
    {
        f=read();
        if(f==1)
        {
            root[i]=root[i-1];
            a=read();b=read();
            int p=find(root[i],a),q=find(root[i],b);
            if(v[p]==v[q])continue;
            if(deep[p]>deep[q])swap(p,q);
            modify(1,n,root[i-1],root[i],v[p],v[q]);
            if(deep[p]==deep[q])add(root[i],1,n,v[q]);
        }
        if(f==2)
        {k=read();root[i]=root[k];}
        if(f==3)
        {
            root[i]=root[i-1];
            a=read();b=read();
            int p=find(root[i],a),q=find(root[i],b);
            if(v[p]==v[q])puts("1");
            else puts("0");
        }
    }
    return 0;
}