HYSBZ 3224 Tyvj 1728 普通平衡树（FHQ_Treap平衡树）

#include<bits/stdc++.h>
using namespace std;
#define ls(x) ch[x][0]
#define rs(x) ch[x][1]
const int maxn=1e5+10;
int tot,root;
int ch[maxn][2],val[maxn],dat[maxn],sz[maxn];
void upd(int x){
    sz[x]=sz[ch[x][0]]+sz[ch[x][1]]+1;
}
int cre(int v){
    ++tot;
    val[tot]=v,sz[tot]=1,dat[tot]=rand();
    return tot;
}
void split(int x,int v,int &tx,int &ty){
    if(!x) tx=ty=0;
    else{
        if(val[x]<=v){
            tx=x;split(ch[x][1],v,ch[x][1],ty);
        }else{
            ty=x;split(ch[x][0],v,tx,ch[x][0]);
        }
        upd(x);
    }
}
int merge(int x,int y){
    if(!x||!y) return x+y;
    if(dat[x]>dat[y]){
        ch[x][1]=merge(ch[x][1],y);
        upd(x);
        return x;
    }else{
        ch[y][0]=merge(x,ch[y][0]);
        upd(y);
        return y;
    }
}
int x,y,z;
inline void ins(int v){
    split(root,v,x,y);
    root=merge(merge(x,cre(v)),y);
}
inline void del(int v){
    split(root,v,x,z);
    split(x,v-1,x,y);
    y=merge(ch[y][0],ch[y][1]);
    root=merge(x,merge(y,z));
}
inline void getrank(int v){
    split(root,v-1,x,y);
    printf("%d\n",sz[x]+1);
    merge(x,y);
}
inline int getval(int now,int v){
    ///cout<<v<<endl;
    if(sz[ch[now][0]]+1<v) return getval(ch[now][1],v-sz[ch[now][0]]-1);
    else if(v==sz[ch[now][0]]+1) return val[now];
    return getval(ch[now][0],v);
}
inline void getpre(int v){
    split(root,v-1,x,y);
    z=x;while(ch[z][1]) z=ch[z][1];
    printf("%d\n",val[z]);
    merge(x,y);
}
inline void getnxt(int v){
    split(root,v,x,y);
    z=y;while(ch[z][0]) z=ch[z][0];
    printf("%d\n",val[z]);
    merge(x,y);
}

int main(){
    ///插入操作
    int cnt;scanf("%d",&cnt);
    while(cnt--){
        int opt,v;
        scanf("%d%d",&opt,&v);
        if(opt==1) ins(v);
        else if(opt==2) del(v);
        else if(opt==3) getrank(v);
        else if(opt==4) printf("%d\n",getval(root,v));
        else if(opt==5) getpre(v);
        else getnxt(v);
    }
    return 0;
}