bzoj 3673&3674: 可持久化并查集 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

题解：

首先普通并查集操作中，是靠fa这个数组维护的

所以要可持久并查集，就要可持久化fa数组，所以用到可持久化线段树维护.

注意要路径压缩

 1 #include <algorithm>
 2 #include <iostream>
 3 #include <cstdlib>
 4 #include <cstring>
 5 #include <cstdio>
 6 #include <cmath>
 7 #define RG register
 8 #define il inline
 9 using namespace std;
10 const int N=2000005,M=20005;
11 int gi(){
12     int str=0;char ch=getchar();
13     while(ch>'9' || ch<'0')ch=getchar();
14     while(ch>='0' && ch<='9')str=(str<<1)+(str<<3)+ch-48,ch=getchar();
15     return str;
16 }
17 int n,m,tot=0,root[M];
18 struct node{
19     int ls,rs,fa;
20 }t[N];
21 il void build(int &rt,int l,int r){
22     rt=++tot;
23     if(l==r){
24         t[rt].fa=l;
25         return ;
26     }
27     int mid=(l+r)>>1;
28     build(t[rt].ls,l,mid);build(t[rt].rs,mid+1,r);
29 }
30 il int query(int rt,int l,int r,int sa){
31     if(l==r)
32         return rt;
33     int mid=(l+r)>>1;
34     if(sa>mid)return query(t[rt].rs,mid+1,r,sa);
35     else return query(t[rt].ls,l,mid,sa);
36 }
37 il void updata(int &rt,int last,int l,int r,int sa,int to){
38     rt=++tot;t[rt]=t[last];
39     if(l==r){
40         t[rt].fa=to;
41         return ;
42     }
43     int mid=(l+r)>>1;
44   if(sa>mid)updata(t[rt].rs,t[last].rs,mid+1,r,sa,to);
45     else updata(t[rt].ls,t[last].ls,l,mid,sa,to);
46 }
47 il int find(int x,int i){
48     int c=query(root[i],1,n,x);
49     if(t[c].fa==x)return x;
50     int d=find(t[c].fa,i);
51     updata(root[i],root[i],1,n,x,d);
52     return d;
53 }
54 void work()
55 {
56     int flag,x,y,fx,fy;
57     n=gi();m=gi();
58     build(root[0],1,n);
59     for(int i=1;i<=m;i++){
60         flag=gi();x=gi();
61         if(flag==1){
62             y=gi();
63             root[i]=root[i-1];
64             fx=find(x,i);fy=find(y,i);
65             if(fx==fy)continue;
66             updata(root[i],root[i-1],1,n,fx,fy);
67         }
68         else if(flag==2)root[i]=root[x];
69         else{
70             y=gi();
71             root[i]=root[i-1];
72             fx=find(x,i);fy=find(y,i);
73             if(fx==fy)puts("1");
74             else puts("0");
75         }
76     }
77 }
78 
79 int main()
80 {
81     work();
82     return 0;
83 }