Treap是一种动态树,其主要操作是旋转rotate,不过我比较喜欢splay(附上链接可以看看哦kkkkk),而这样就不能支持持久化了
为了强行优化这一点,我们决定将其转化为静态的Treap,每插入一个新的数据,都先复制上一个版本,再进行修改,这样会极大的
节省空间。我们采取使用随机权值fix来强行修改树的结构,降低树的深度,变成了log,这样,就在持久化的同时进行了优化
#include<iostream> #include<cstdio> #include<string> #include<algorithm> #include<string> #define lc(x) ch[x][0] #define rc(x) ch[x][1] #define N 300010 #define M (N*50) using namespace std; typedef pair <int ,int > dat; int ch[M][2],val[M],siz[M],fix[M];//val维护BST的性质,fix维护斜堆的性质 int cnt=0,n,rt[N]; +
inline int Read() { int num=0,k=1; char c=getchar(); while(c!='-'&&(c<'0'||c>'9')) c=getchar(); if(c=='-'){k=-1;c=getchar();} while(c>='0'&&c<='9'){num=(num<<1)+(num<<3)+(c^48);c=getchar();} return num*k; } inline void Copy (int a,int b) { val[a]=val[b];siz[a]=siz[b]; ch[a][0]=ch[b][0];ch[a][1]=ch[b][1]; fix[a]=fix[b]; } inline void push_up(int o) { siz[o]=siz[lc(o)]+siz[rc(o)]+1; } inline int New_node (int x=0) +
{ val[++cnt]=x; siz[cnt]=1; ch[cnt][0]=ch[cnt][1]=0; fix[cnt]=rand()%5000; return cnt; } inline int merge(int a,int b)//利用fix强行维护堆的性质,判断依据是其fix值 { //但实际上我们同时维护了BST的性质,这就是为什么合并到左右子树会有区别 if(!a || !b) return a+b; int x=++cnt; if(fix[a]<fix[b]) { Copy(x,a); ch[x][1]=merge(ch[x][1],b); } else { Copy(x,b); ch[x][0]=merge(a,ch[x][0]); } push_up(x); return x; } inline dat split(int o,int k)//分裂 { if(!o) return make_pair(0,0); int x=++cnt; Copy(x,o); dat t; if(k<=siz[lc(x)]) { t=split(lc(x),k); ch[x][0]=t.second; push_up(x); return make_pair(t.first,x); } else { t=split(rc(x),k-siz[lc(x)]-1); ch[x][1]=t.first; push_up(x); return make_pair(x,t.second); } } int insert(int o,int k,int x) { dat t=split(o,k-1); int tmp=New_node(x); int ret=merge(merge(t.first,tmp),t.second); return ret; } int del(int o,int k) { dat a=split(o,k-1); dat b=split(a.second,1); return merge(a.first,b.second); } int No_k(int o,int k) { if(!o)return 0; if(k==siz[lc(o)]+1) return val[o]; if(k<=siz[lc(o)])return No_k(lc(o),k); return No_k(rc(o),k-siz[lc(o)]-1); } void dfs(int o) { if(!o) return; dfs(lc(o)); printf("%d ",val[o]); dfs(rc(o)); } int main() { int tot=0; rt[0]=0; n=Read(); for(int i=1;i<=n;i++) { int op,t,k,x; op=Read();t=Read();k=Read(); if(op==1) x=Read(); switch(op) { case 1:{rt[++tot]=insert(rt[t],k,x);break;} case 2:{rt[++tot]=del(rt[t],k);break;} case 3: {printf("%d\n",No_k(rt[t],k));break;} } } return 0; }