HYSBZ 3224 Tyvj 1728 普通平衡树

#include<bits/stdc++.h>
using namespace std;
#define ls ch[x][0]
#define rs ch[x][1]
#define R register int
const int maxn=1e5+10,Inf=2e9+10;
inline int input(){
    R ret=0,fix=1;
    register char ch;
    while(!isdigit(ch=getchar())) fix=(ch=='-')?-1:fix;
    do ret=ret*10+(ch^48);
    while(isdigit(ch=getchar()));
    return ret*fix;
}
int n,tot,rt;
int sz[maxn],ch[maxn][2],val[maxn],dat[maxn],cnt[maxn];
inline void upd(int x){
    sz[x]=sz[ls]+sz[rs]+cnt[x];
}
inline int cre(int v){
    R x=++tot;
    cnt[x]=1,val[x]=v,dat[x]=rand(),upd(x);
    return tot;
}
void rot(int& x,int d){
    R y=ch[x][d];
    ch[x][d]=ch[y][d^1];
    ch[y][d^1]=x;
    upd(x),upd(y);
    x=y;
}
void ins(int& x,int v){
    if(!x) {x=cre(v);return;}
    if(val[x]==v){
        ++cnt[x];upd(x);
        return ;
    }
    R d=val[x]<v;
    ins(ch[x][d],v);
    upd(x);
    if(dat[ch[x][d]]<dat[x]) rot(x,d);
}
void del(int& x,int v){
    if(!x) return;
    if(val[x]==v){
        if(cnt[x]>1) cnt[x]--;
        else{
            if(!ls) x=rs;
            else if(!rs) x=ls;
            else{
                R d=dat[ls]>dat[rs];
                rot(x,d);
                del(ch[x][d^1],v);
            }
        }
    }else del(ch[x][val[x]<v],v);
    upd(x);
}
void build(){
    srand(100023323);
    rt=cre(-Inf);
    ins(rt,Inf);
}
int getpre(int x,int v){
    if(!x) return -Inf;
    if(val[x]<v) return max(getpre(rs,v),val[x]);
    else return getpre(ls,v);
}
int getnxt(int x,int v){
    if(!x) return Inf;
    if(val[x]>v) return min(getnxt(ls,v),val[x]);
    else return getnxt(rs,v);
}
int getrk(int x,int v){
    if(!x) return 0;
    if(val[x]==v) return sz[ls]+1;
    else if(val[x]>v) return getrk(ls,v);
    else return sz[ls]+cnt[x]+getrk(rs,v);
}
int getval(int x,int rk){
    if(!x||!rk) return 0;
    if(rk<=sz[ls]) return getval(ls,rk);
    else if(rk<=sz[ls]+cnt[x]) return val[x];
    return getval(rs,rk-sz[ls]-cnt[x]);
}
int main(){
    build();
    int tott=input();
    while(tott--){
        int opt=input(),x=input();
        if(opt==1) ins(rt,x);
        else if(opt==2) del(rt,x);
        else if(opt==3) printf("%d\n",getrk(rt,x)-1);
        else if(opt==4) printf("%d\n",getval(rt,x+1));
        else if(opt==5) printf("%d\n",getpre(rt,x));
        else printf("%d\n",getnxt(rt,x));
    }
    return 0;
}