bzoj 3224 平衡树模板题 （FHQ treap模板）

自己撸的FHQ treap 基本板子， 虽然是权值平衡树 ，但是好像也没有用权值split， 还是用的siz_split。
存一下板子。

代码

#include<bits/stdc++.h>
using namespace std;
#define maxn 200005
#define maxm 1006
#define ll long long int
#define INF 0x3f3f3f3f
#define inc(i,l,r) for(int i=l;i<=r;i++)
#define dec(i,r,l) for(int i=r;i>=l;i--)
#define mem(a) memset(a,0,sizeof(a))
#define sqr(x) (x*x)
#define inf (ll)2e18+1
#define PI acos(-1)
#define mod 10007
#define auto(i,x) for(int i=head[x];i;i=ed[i].nxt)
ll read(){
    ll x=0,f=1ll;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
     while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
     return f*x;
}
int n;
int val[maxn],siz[maxn],rnd[maxn],ls[maxn],rs[maxn],tot,rt;
void pushup(int now){siz[now]=siz[ls[now]]+siz[rs[now]]+1; }///记得加自己...
void split(int now,int k,int &x,int &y){
    if(now==0){x=y=0;return ; }
    if(siz[ls[now]]>=k){
        y=now;
        split(ls[now],k,x,ls[now]);
    }
    else{
        x=now;
        split(rs[now],k-siz[ls[now]]-1,rs[now],y);
    }
    pushup(now);
}
void Merge(int &now,int x,int y){
    if(!x||!y){now=x+y;return ;}
    if(rnd[x]<rnd[y]){
        now=x;
        Merge(rs[now],rs[now],y);
    }
    else{
        now=y;
        Merge(ls[now],x,ls[now]);
    }
    pushup(now);
}
int getrank(int va){///返回严格小于x的数的个数
    int res=0,mi=INF,now=rt;
    while(now){
        if(va<=val[now]){
            if(va==val[now]){
                mi=min(mi,res+siz[ls[now]]);///这个地方比较精髓因为 原树可以有多个相同的权值的点
            }                               ///所以要找到一个最小的
            now=ls[now];
        }
        else {
            res+=siz[ls[now]]+1;
            now=rs[now];
        }
    }
    return mi==INF?res:mi;
}
void insert(int va){
    int pos=getrank(va);
    int x=0,y=0,z=0;
    split(rt,pos,x,y);
    z=++tot;
    rnd[z]=rand();val[z]=va;siz[z]=1;
    Merge(rt,x,z);
    Merge(rt,rt,y);
}
void del(int va){
    int pos=getrank(va);
    int x=0,y=0,z=0;
    split(rt,pos,x,y);
    split(y,1,y,z);
    if(val[y]==va)Merge(rt,x,z);
}
int pre(int va){
    int pos=getrank(va);
    int x=0,y=0,z=0;
    split(rt,pos,x,z);
    split(x,pos-1,x,y);
    int res=val[y];
    Merge(rt,x,y);
    Merge(rt,rt,z);
    return res;
}
int nxt(int va){
    int pos=getrank(va+1);
    int x=0,y=0,z=0;
    split(rt,pos,x,z);
    split(z,1,y,z);
    int res=val[y];
    Merge(rt,y,z);
    Merge(rt,x,rt);
    return res;
}
int getnum(int pos){
    int x=0,y=0,z=0;
    split(rt,pos,x,z);
    split(x,pos-1,x,y);
    int res=val[y];
    Merge(rt,x,y);
    Merge(rt,rt,z);
    return res;
}
/*void dfs(int x){
    if(val[x]==0){printf("*\n");return ; }
    printf("val = %d    siz = %d\n",val[x],siz[x]);
    dfs(ls[x]);
    dfs(rs[x]);
    printf("val = %d    siz = %d\n",val[x],siz[x]);
}*/
int main()
{
    srand(19260817);
    n=read();
    int op,x;
    inc(i,1,n){
        op=read();x=read();
        if(op==1)insert(x);
        else if(op==2)del(x);
        else if(op==3)printf("%d\n",getrank(x)+1);
        else if(op==4)printf("%d\n",getnum(x));
        else if(op==5)printf("%d\n",pre(x));
        else if(op==6)printf("%d\n",nxt(x));
    }
    return 0;
}