[模板]平衡树treap

#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<ctime>
#include<queue>
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]]<rnd[k])lturn(k);
	 }
	 else{
		 insert(l[k],x);
		 if(rnd[l[k]]<rnd[k])rturn(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]]<rnd[r[k]]){rturn(k);del(k,x);}
		 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]<x){
		 ans=k;query_pro(r[k],x);
	 }
     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 main (){
    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;
}