[模板]平衡树treap

#include
#include
#include
#include
#include
#include
#include
#define LiangJiaJun main
using namespace std;
int l[1000004],r[1000004],s[1000004],rnd[1000004],w[1000004],v[1000004];
int n,sz,root,ans;
void update(int k){
     s[k]=s[l[k]]+s[r[k]]+w[k];
}
void rturn(int &k){
     int t=l[k];l[k]=r[t];r[t]=k;s[t]=s[k];
     update(k);k=t;
}
void lturn(int &k){
     int t=r[k];r[k]=l[t];l[t]=k;s[t]=s[k];
     update(k);k=t;
}
void insert(int &k,int x){//插入操作
     if(k==0){
         k=++sz;
         s[k]=w[k]=1;v[k]=x;rnd[k]=rand();
         return;
     }
     s[k]++;
     if(v[k]==x)w[k]++;
     else if(x>v[k]){
         insert(r[k],x);
         if(rnd[r[k]]else{
         insert(l[k],x);
         if(rnd[l[k]]void del(int &k,int x){//删除操作
     if(k==0)return;
     if(v[k]==x){
         if(w[k]>1){
             w[k]--;s[k]--;return;
         }
         if(l[k]*r[k]==0)k=l[k]+r[k];
         else if(rnd[l[k]]else {lturn(k);del(k,x);}
     }
     else{
         s[k]--;
         if(x>v[k])del(r[k],x);
         else del(l[k],x);
     }
}
int query_num(int k,int x){//查找数字x是第几小
    if(k==0)return 0;
    if(v[k]==x)return s[l[k]]+1;
    else if(x>v[k])return s[l[k]]+w[k]+query_num(r[k],x);
    else return query_num(l[k],x);
}
int query_rank(int k,int x){//查找第x小的数
    if(k==0)return 0;
    if(x<=s[l[k]])return query_rank(l[k],x);
    else if(x>s[l[k]]+w[k])return query_rank(r[k],x-s[l[k]]-w[k]);
    else return v[k];
}
void query_pro(int k,int x){//查x的前驱
     if(k==0)return;
     if(v[k]else query_pro(l[k],x);
}
void query_sub(int k,int x){//查x的后继
     if(k==0)return;
     if(v[k]>x){
        ans=k;query_sub(l[k],x);
     }
     else query_sub(r[k],x);
}
int LiangJiaJun (){
    srand(time(NULL));
    scanf("%d",&n);
    int opt,x;
    for(int i=1;i<=n;i++){
        scanf("%d%d",&opt,&x);
        switch(opt){
            case 1:insert(root,x);break;
            case 2:del(root,x);break;
            case 3:printf("%d\n",query_num(root,x));break;
            case 4:printf("%d\n",query_rank(root,x));break;
            case 5:ans=0;query_pro(root,x);printf("%d\n",v[ans]);break;
            case 6:ans=0;query_sub(root,x);printf("%d\n",v[ans]);break;
        }
    }
    return 0;
}