洛谷 P3369 【模板】普通平衡树（Treap/SBT）

题目描述
您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作：
插入x数
删除x数(若有多个相同的数，因只删除一个)
查询x数的排名(若有多个相同的数，因输出最小的排名)
查询排名为x的数
求x的前驱(前驱定义为小于x，且最大的数)
求x的后继(后继定义为大于x，且最小的数)
输入输出格式
输入格式：
第一行为n，表示操作的个数,下面n行每行有两个数opt和x，opt表示操作的序号(1<=opt<=6)
输出格式：
对于操作3,4,5,6每行输出一个数，表示对应答案
输入输出样例
输入样例#1：
10
1 106465
4 1
1 317721
1 460929
1 644985
1 84185
1 89851
6 81968
1 492737
5 493598
输出样例#1：
106465
84185
492737
说明
时空限制：1000ms,128M
1.n的数据范围：n<=100000
2.每个数的数据范围：[-1e7,1e7]
来源：Tyvj1728 原名：普通平衡树
在此鸣谢

//我们用SPlay 来写Treap的模板题    谢徐队的代码参考 
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define MAXN 1000000
int cnt[MAXN],key[MAXN],ch[MAXN][2],fa[MAXN],siz[MAXN];
int sz,root;
inline void UpDate(int u){
    siz[u]=cnt[u];
    if(ch[u][0]) siz[u]+=siz[ch[u][0]];
    if(ch[u][1]) siz[u]+=siz[ch[u][1]];
}
inline int pre(){
    int u=ch[root][0];
    while(ch[u][1]) u=ch[u][1];//宸﹀瀛愮殑鍙冲瀛愮殑鍙?..
    return u;
}
inline int next(){
    int u=ch[root][1];
    while(ch[u][0]) u=ch[u][0];
    return u;
}
inline int getson(int x){ return ch[fa[x]][1]==x; }
inline void clear(int x){
    ch[x][0]=ch[x][1]=cnt[x]=siz[x]=fa[x]=key[x]=0;
}
inline void Roteta(int x){
    int f=fa[x],f_fa=fa[f],k=getson(x);
    ch[f][k]=ch[x][k^1];fa[ch[f][k]]=f;
    ch[x][k^1]=f;fa[f]=x;
    fa[x]=f_fa;
    if(f_fa) ch[f_fa][ch[f_fa][1]==f]=x;
    UpDate(f);UpDate(x);
}
inline void SPlay(int x){
    for(int f;f=fa[x];Roteta(x))
        if(fa[f]) Roteta(getson(x)==getson(f) ? f :x);
    root=x;
}
inline int findrank(int x){//鏌ヨx鐨勬帓鍚?
    int now=root,ans=0;
    while(1){
        if(x<key[now]) now=ch[now][0];
        else{
            ans+=ch[now][0] ? siz[ch[now][0]] : 0;
            if(x==key[now]){
                SPlay(now);
                return ans+1;
            }
            ans+=cnt[now];now=ch[now][1];
        }
    }
}
inline int findx(int x){
    int now=root;
    while(1){
        if(ch[now][0]&&x<=siz[ch[now][0]]) now=ch[now][0];
        else{
            int temp=(ch[now][0] ? siz[ch[now][0]] : 0)+cnt[now];
            if(x<=temp) return key[now];
            x-=temp;
            now=ch[now][1];
        }
    }
}
inline void create(int x){
    sz++;key[sz]=x;
    cnt[sz]=siz[sz]=1;
    ch[sz][0]=ch[sz][1]=fa[sz]=0;
}
void insert(int x){
    if(!root) create(x),root=sz;
    else{
        int now=root,f=0;
        while(1){
            if(key[now]==x){
                cnt[now]++;siz[now]++;
                SPlay(now);break;
            }
            f=now;now=ch[f][x>key[f]];
            if(!now){
                create(x);
                fa[sz]=f;
                ch[f][x>key[f]]=sz;
                SPlay(sz);
                break;
            }
        }
    }
}
inline void del(int x){
    int t=findrank(x);//x鐨勬帓鍚?
    if(cnt[root]>1){
        cnt[root]--;siz[root]--;
        return ;
    }
    if(!ch[root][0]&&!ch[root][1]){ clear(root);root=0;return; }
    if(!ch[root][0]){
        int cur=root;
        root=ch[root][1];
        fa[root]=0;
        clear(cur);
        return;
    }
    if(!ch[root][1]){
        int tmp=root;
        root=ch[root][0];
        fa[root]=0;
        clear(tmp);
        return;
    }
    int pre1=pre(),tmp=root;
    SPlay(pre1);
    ch[root][1]=ch[tmp][1];
    fa[ch[tmp][1]]=root;
    clear(tmp);
    UpDate(root);
}

int main(){
    int n,opt,x;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d%d",&opt,&x);
        switch(opt){
            case 1:insert(x);break;
            case 2:del(x);break;
            case 3:printf("%d\n",findrank(x));break;//x鐨勬帓鍚?
            case 4:printf("%d\n",findx(x));break;//鎺掑悕涓簒鐨?
            case 5:insert(x);printf("%d\n",key[pre()]);del(x);break;
            case 6:insert(x);printf("%d\n",key[next()]);del(x);break;
        }
    }
    return 0;
}