BZOJ 3673: 可持久化并查集 by zky

BZOJ 3673: 可持久化并查集 by zky

标签（空格分隔）： OI-BZOJ OI-可持久化数组 OI-并查集

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

Solution

可持久化数组：用可持久化线段树来储存数组的值，
对变量的查询相当于对线段树某点修改查询，实现的
时候可以在把数值存在线段树的l,r指针内，因为只有
叶节点有数值，叶节点没有孩子，利用空间。
可持久化并查集：数组可持久化
st[time]表示time时刻结束后的线段树根节点
w[0]左孩子,w[1]右孩子

听说启发式合并暴力修改啥的似乎也能过。

Code

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<queue>
#include<set>
#include<map>
#include<bitset>
#include<vector>
using namespace std;
#define PA pair<int,int>
int read()
{
    int s=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){s=(s<<1)+(s<<3)+ch-'0';ch=getchar();}
    return s*f;
}
//smile please
struct tree;
extern tree t[5000005];
struct tree
{
    int w[2];
}t[5000005];
int n,m,c,np,timee,la;
int st[20005];
int ask(int x,int s,int c)
{
    if(c==-1)return t[x].w[0];
    int p=(s>>c)&1;
    return ask(t[x].w[p],s-(p<<c),c-1);
}
void modif(int x,int s,int c,int num)
{
    int now=++np;
    t[now]=t[x];
    if(c==-1){t[now].w[0]=num;return;}
    int p=(s>>c)&1;
    modif(t[now].w[p],s-(p<<c),c-1,num);
    t[now].w[p]=now+1;
}
int gf(int x)
{
    int fx=ask(st[timee],x,c-1);
    if(fx==x)
       return x;
    else
      {fx=gf(fx);
       la=np+1;
       modif(st[timee],x,c-1,fx);
       st[timee]=la;
       return fx;
      }
}
int main()
{
    n=read(),m=read();
    c=ceil(log(n)/log(2));
    for(int i=1;i<=n;i++)
        la=np+1,
        modif(st[0],i,c-1,i),
        st[0]=la;
    for(timee=1;timee<=m;timee++)
       {int o=read();
        if(o==2)
           st[timee]=st[read()];
        else
           st[timee]=st[timee-1];
        if(o==1)
          {int a=read(),b=read();
           gf(a),gf(b);
           int fa=ask(st[timee],a,c-1),
               fb=ask(st[timee],b,c-1);
           if(fa!=fb)
              la=np+1,
              modif(st[timee],fa,c-1,fb),
              st[timee]=la;
          }
        if(o==3)
          {int a=read(),b=read();
           gf(a),gf(b);
           int fa=ask(st[timee],a,c-1),
               fb=ask(st[timee],b,c-1);
           printf("%d\n",fa==fb);
          }
       }
    return 0;
}