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

→题目链接←

【想说的话】

rope大法好！！！！！

【题解】

用rope来实现可持久化并查集

rp[i]->at(x)                                 访问下标为x的元素的值

rp[i]->replace(pos,x)                 将下标为pos的元素的值更改为x

rp[i]=new rope<int>(*rp[y])       将rp[y]整个复制到rp[i]

【代码】

#include<bits/stdc++.h>
#include<ext/rope>

#define MAXN 20020

using namespace std;
using namespace __gnu_cxx;

inline int rd(){
int x=0,y=1;char c=getchar();
while(c<'0' || c>'9'){if(c=='-')y=-y;c=getchar();}
while(c>='0'&&c<='9')x=x*10+c-'0',c=getchar();
return x*y;
}

int n,m;
rope<int> *rp[MAXN];
int a[MAXN];

inline int fa(int i,int x){
return rp[i]->at(x);
}

int getroot(int i,int x){
if(fa(i,x)==x)return x;
rp[i]->replace(x,getroot(i,fa(i,x)));
return fa(i,x);
}

inline void un(int i,int x,int y){
x=getroot(i,x);
y=getroot(i,y);
if(x==y)return;
rp[i]->replace(x,y);
}

int main(){
n=rd(),m=rd();
for(int i=1; i<=n; i++)a[i]=i;
rp[0]=new rope<int>(a,a+n+1);
int i=0;
while(m--){
i++;
rp[i]=new rope<int>(*rp[i-1]);
int x=rd(),y,z;
if(x==1)y=rd(),z=rd(),un(i,y,z);
else if(x==2)y=rd(),rp[i]=new rope<int>(*rp[y]);
else y=rd(),z=rd(),printf("%d\n",getroot(i,y)==getroot(i,z));
}

return 0;
}