BZOJ 3674 可持久化并查集加强版(按秩合并版本)

 /*
 bzoj 3674: 可持久化并查集加强版
 http://www.lydsy.com/JudgeOnline/problem.php?id=3674
 用可持久化线段树维护可持久化数组从而实现可持久化并查集
 可持久化线段树+并查集+按秩合并+读入优化
 */
 #include <cstdio>
 #include <algorithm>
 using namespace std;
 const int Nmax=;
 int root_fa[Nmax],root_rankk[Nmax];

 inline int read()
 {
     int x=;char ch=getchar();
     while(ch>''||ch<'')ch=getchar();
     while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
     return x;
 }
 struct Node
 {
     int ls;
     int rs;
     int data;
 };
 Node fa[*Nmax];
 Node rankk[*Nmax];
 int n,m,ans,a,b,k;
 int size_fa,size_rankk;

 void build_rankk(int &root,int l,int r)
 {
     root=size_rankk++;
     int mid=(l+r)>>;
     if(l==r)
     {
         rankk[root].data=;
         return;
     }
     build_rankk(rankk[root].ls,l,mid);
     build_rankk(rankk[root].rs,mid+,r);
 }

 void insert_rankk(int last,int &root,int pos,int l,int r)
 {
     rankk[size_rankk]=rankk[last];
     root=size_rankk;
     size_rankk++;
     if(rankk[root].ls==rankk[root].rs)
     {
         rankk[root].data++;
         return;
     }
     int mid=(l+r)>>;
     if(pos<=mid)
         insert_rankk(rankk[last].ls,rankk[root].ls,pos,l,mid);
     else
         insert_rankk(rankk[last].rs,rankk[root].rs,pos,mid+,r);
 }

 int search_rankk(int root,int pos,int l,int r)
 {
     if(rankk[root].ls==rankk[root].rs)
         return rankk[root].data;
     int mid=(l+r)>>;
     if(pos<=mid)
         return search_rankk(rankk[root].ls,pos,l,mid);
     else
         return search_rankk(rankk[root].rs,pos,mid+,r);
 }

 void build_fa(int &root,int l,int r)
 {
     root=size_fa++;
     int mid=(l+r)>>;
     if(l==r)
     {
         fa[root].data=l;
         return;
     }
     build_fa(fa[root].ls,l,mid);
     build_fa(fa[root].rs,mid+,r);
 }

 void insert_fa(int last,int &root,int pos,int data,int l,int r)
 {
     fa[size_fa]=fa[last];
     root=size_fa;
     size_fa++;
     if(fa[root].ls==fa[root].rs)
     {
         // printf("root:%d,l:%d,data:%d\n",root,fa[root].l,data);
         fa[root].data=data;
         return;
     }
     int mid=(l+r)>>;
     if(pos<=mid)
         insert_fa(fa[last].ls,fa[root].ls,pos,data,l,mid);
     else
         insert_fa(fa[last].rs,fa[root].rs,pos,data,mid+,r);
 }

 int search_fa(int root,int pos,int l,int r)
 {
     if(fa[root].ls==fa[root].rs)
         return fa[root].data;
     int mid=(l+r)>>;
     if(pos<=mid)
         return search_fa(fa[root].ls,pos,l,mid);
     else
         return search_fa(fa[root].rs,pos,mid+,r);
 }

 int find(int root,int x)
 {
     // if(x==0)
     //     printf("error!!!!\n");
     int fa=search_fa(root,x,,n);
     if(fa!=x)
         return find(root,fa);
     else
         return fa;
 }

 // void watch(int root,int l,int r)
 // {
 //     printf("tree[%d].l=%d,tree[%d].r=%d,tree[%d].data=%d\n",root,fa[root].l,root,fa[root].r,root,fa[root].data);
 //     if(l==r)
 //         return;
 //     int mid=(l+r)>>1;
 //     watch(fa[root].ls,l,mid);
 //     watch(fa[root].rs,mid+1,r);
 // }

 int main()
 {
     freopen("bzoj3674.in","r",stdin);
     //scanf("%d%d",&n,&m);
     n=read();
     m=read();
     ans=;
     int q;
     build_fa(root_fa[],,n);
     build_rankk(root_rankk[],,n);
     // printf("the 0 watch:\n");
     //     watch(root_fa[0],1,n);
     for(int i=;i<=m;i++)
     {
         //scanf("%d",&q);
         q=read();
         if(q==)
         {
             a=read();
             b=read();
             // scanf("%d%d",&a,&b);
             a^=ans;b^=ans;
             int rt1=find(root_fa[i-],a),rt2=find(root_fa[i-],b);
             // printf("root[%d]=%d,root[%d]=%d\n",a,rt1,b,rt2);
             if(rt1==rt2)
             {
                 root_fa[i]=root_fa[i-];
                 root_rankk[i]=root_rankk[i-];
             }
             else
             {
                 int rk1=search_rankk(root_rankk[i-],rt1,,n),rk2=search_rankk(root_rankk[i-],rt2,,n);
                 if(rk1<rk2)
                 {
                     insert_fa(root_fa[i-],root_fa[i],rt1,rt2,,n);
                     root_rankk[i]=root_rankk[i-];
                 }
                 else
                 {
                     insert_fa(root_fa[i-],root_fa[i],rt2,rt1,,n);
                     if(rk1==rk2)
                         insert_rankk(root_rankk[i-],root_rankk[i],rt1,,n);
                     else
                         root_rankk[i]=root_rankk[i-];
                 }
             }
             // printf("search_fa[2]=%d\n",search_fa(root_fa[1],2) );
             // printf("root[%d]=%d,root[%d]=%d\n",a,find(root_fa[i],a),b,find(root_fa[i],b));
         }
         else if(q==)
         {
             k=read();
             // scanf("%d",&k);
             k^=ans;
             root_fa[i]=root_fa[k];
             root_rankk[i]=root_rankk[k];
         }
         else if(q==)
         {
             a=read();
             b=read();
             // scanf("%d%d",&a,&b);
             root_fa[i]=root_fa[i-];
             root_rankk[i]=root_rankk[i-];
             a^=ans;
             b^=ans;
             if(find(root_fa[i],a)==find(root_fa[i],b))
                 ans=;
             else
                 ans=;
             printf("%d\n",ans);
         }
         //insert_fa(root_fa[i-1],root_fa[i],2,3);
         // printf("root_fa[%d]:%d\n",i,root_fa[i]);
         // printf("the %d watch:\n",i);
         // watch(root_fa[i],1,n);
     }
     return ;
 }