【模板】可持久化Treap

洛谷3835

 #include<cstdio>
 #include<algorithm>
 #include<cstdlib>
 #define ls (a[u].l)
 #define rs (a[u].r)
 #define R (root[Ver])
 #define update(u) (a[u].size=a[a[u].l].size+a[a[u].r].size+1)
 #define copy(x) (a[++tot]=a[x],a[x=tot].ver=Ver)
 using namespace std;
 int Ver,ver,Opt,Val,n,x,y,z,tot,root[];
 struct treap{int l,r,val,rnd,size,ver;}a[];
 inline void read(int &k){
     k=; int f=; char c=getchar();
     while(c<''||c>'')c=='-'&&(f=-),c=getchar();
     while(''<=c&&c<='')k=k*+c-'',c=getchar();
     k*=f;
 }
 inline void put(int x){
     if(x<) putchar('-'),x=-x;
     char s[]; int k=,y=;
     while(x>){
         y=x; x/=;
         s[++k]=y-x*+;
     }
     for (int i=k;i>=;i--) putchar(s[i]);
     puts(y?"":"");
 }
 inline void newnode(int val){a[++tot]=(treap){,,val,rand(),,Ver};}
 void split(int u,int k,int &x,int &y){
     if(!k){x=; y=u; return;}
     if(a[u].size==k){x=u; y=; return;}
     if (a[u].ver<Ver) copy(u);
     if(a[ls].size>=k) split(ls,k,x,ls),y=u;
     else split(rs,k-a[ls].size-,rs,y),x=u;
     update(u);
 }
 int merge(int x,int y){//x较小树，y较大树
     if ((!x)||(!y)) return x+y;
     if(a[x].rnd<a[y].rnd){if (a[x].ver<Ver) copy(x); a[x].r=merge(a[x].r,y); update(x); return x;}
     else{if (a[y].ver<Ver) copy(y); a[y].l=merge(x,a[y].l); update(y); return y;}
 }
 int qrank(int u,int val){
     if(!u) return ;
     return a[u].val>=val?qrank(ls,val):qrank(rs,val)+a[ls].size+;
 }
 int qval(int u,int rank){
     if(a[ls].size+==rank) return a[u].val;
     return a[ls].size>=rank?qval(ls,rank):qval(rs,rank-a[ls].size-);
 }
 int main(){
     srand(); a[root[]=tot=]=(treap){,,<<,-,,};
     read(n);
     for(Ver=;Ver<=n;Ver++){
         read(ver); R=root[ver];
         read(Opt); read(Val);
         if(Opt==){split(R,qrank(R,Val),x,y); newnode(Val); R=merge(merge(x,tot),y);}//插入
         if(Opt==){//删除
             int tmp=qrank(R,Val);
             if (qval(R,tmp+)!=Val) continue;
             split(R,tmp,x,y); split(y,,z,y); R=merge(x,y);
         }
         if(Opt==) put(qrank(R,Val)+);//求x的排名
         if(Opt==) put(qval(R,Val));//求排名为x的数
         if(Opt==){//求x的前驱
             int tmp=qrank(R,Val);
             if(tmp) put(qval(R,tmp)); else puts("-2147483647");
         }
         if(Opt==){//求x的后继
             int tmp=qrank(R,Val+);
             if(tmp<a[R].size) put(qval(R,tmp+)); else puts("");
         }
     }
     return ;
 }