【bzoj3673】可持久化并查集 by zky

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

题解

对每一个时刻建立一颗可持久化线段树，维护信息。

代码

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
int n,m,sz;
int root[200005],ls[2000005],rs[2000005],v[2000005],deep[2000005];
inline 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-'0';ch=getchar();}
return x*f;
}
void build(int &k,int l,int r)
{
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 opt,k,a,b;
for (int i=1;i<=m;i++)
{
opt=read();
if (opt==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]);
}
else if (opt==2)
{
k=read();root[i]=root[k];
}
else
{
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;
}