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;
}