Treap静态非指针模板----BZOJ3224:Tyvj 1728 普通平衡树

https://www.lydsy.com/JudgeOnline/problem.php?id=3224

这道题就是一个Treap模板。

大家初学Treap的时候遇到的Treap都是指针的，有些同学看着费劲而且有时候不好写，就跟我当初学习AC自动机一样，网上的模板都是带有指针的，我看着很不习惯，然后偶然在B站看到可以直接用数组模拟写，然后我本能的发现这个一看就一目了然。这次也找来了静态的模板。

这次就直接全部给大家，就不一部分一部分的写了。

其中定义inf要注意，防止溢出。其中dat随机旋转，是防止退化成一条链，影响复杂度。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
namespace io {
    const int SIZE = 1e7 + 10;
    char inbuff[SIZE];
    char *l, *r;
    inline void init() {
        l = inbuff;
        r = inbuff + fread(inbuff, 1, SIZE, stdin);
    }
    inline char gc() {
        if (l == r) init();
        return (l != r) ? *(l++) : EOF;
    }
    void read(int &x) {
        x = 0; char ch = gc();
        while (!isdigit(ch)) ch = gc();
        while (isdigit(ch)) x = x * 10 + ch - '0', ch = gc();
    }
} using io::read;
const int inf=INT_MAX;
const int N=1e5+10;
const int M=20;
const ll mod=1000000;
struct Treap
{
    int l,r;///左右儿子
    int dat,val;///随机值，价值
    int cnt,sz;///重复数，子树大小
}t[N];
int tot,root,n;
void init(){tot=0;}
int newnode(int val)
{
    t[++tot].val=val;t[tot].dat=rand();
    t[tot].l=t[tot].r=0;
    t[tot].cnt=t[tot].sz=1;
    return tot;
}
void update(int p)
{
    t[p].sz=t[t[p].l].sz+t[t[p].r].sz+t[p].cnt;
}
void build()
{
    newnode(-inf);
    newnode(inf);
    root=1,t[1].r=2;
    update(root);
}
void zig(int &p)
{
    int q=t[p].l;
    t[p].l=t[q].r,t[q].r=p,p=q;
    update(t[p].r),update(p);
}
void zag(int &p)
{
    int q=t[p].r;
    t[p].r=t[q].l,t[q].l=p,p=q;
    update(t[p].l),update(p);
}
void Insert(int &p,int val)
{
    if(p==0){
        p=newnode(val);
        return;
    }
    if(val==t[p].val){
        t[p].cnt++,update(p);
        return;
    }
    if(val<t[p].val){
        Insert(t[p].l,val);
        if(t[p].dat<t[t[p].l].dat) zig(p);
    }
    else{
        Insert(t[p].r,val);
        if(t[p].dat<t[t[p].r].dat) zag(p);
    }
    update(p);
}
void Remove(int &p,int val)
{
    if(p==0) return;
    if(val==t[p].val){
        if(t[p].cnt>1){
            t[p].cnt--,update(p);
            return;
        }
        if(t[p].l||t[p].r){
            if(t[p].r==0||t[t[p].l].dat>t[t[p].r].dat)
                zig(p),Remove(t[p].r,val);
            else zag(p),Remove(t[p].l,val);
            update(p);
        }
        else p=0;
        return;
    }
    val<t[p].val?Remove(t[p].l,val):Remove(t[p].r,val);
    update(p);
}

int GetVal(int p,int val){///注意inf
    if(p==0) return 0;
    if(val==t[p].val) return t[t[p].l].sz+1;///注意
    if(val<t[p].val) return GetVal(t[p].l,val);
    return GetVal(t[p].r,val)+t[t[p].l].sz+t[p].cnt;
}

int GetRank(int p,int rk){///注意inf
    if(p==0) return inf;
    if(t[t[p].l].sz>=rk) return GetRank(t[p].l,rk);
    if(t[t[p].l].sz+t[p].cnt>=rk) return t[p].val;
    return GetRank(t[p].r,rk-t[t[p].l].sz-t[p].cnt);
}

int GetPre(int val){
    int ans=1;
    int p=root;
    while(p){
        if(val==t[p].val){
            if(t[p].l>0){
                p=t[p].l;
                while(t[p].r>0) p=t[p].r;
                ans=p;
            }
            break;
        }
        if(t[p].val<val&&t[p].val>t[ans].val) ans=p;
        p=val<t[p].val?t[p].l:t[p].r;
    }
    return t[ans].val;
}
int GetNext(int val){
    int ans=2;
    int p=root;
    while(p){
        if(val==t[p].val){
            if(t[p].r>0){
                p=t[p].r;
                while(t[p].l>0) p=t[p].l;
                ans=p;
            }
            break;
        }
        if(t[p].val>val&&t[p].val<t[ans].val) ans=p;
        p=val<t[p].val?t[p].l:t[p].r;
    }
    return t[ans].val;
}

int main()
{
    scanf("%d",&n);
    int opt,x;
    build();
    for(int i=1;i<=n;i++){
        scanf("%d%d",&opt,&x);
        if(opt==1) Insert(root,x);
        else if(opt==2) Remove(root,x);
        else if(opt==3) printf("%d\n",GetVal(root,x)-1);
        else if(opt==4) printf("%d\n",GetRank(root,x+1));
        else if(opt==5) printf("%d\n",GetPre(x));
        else printf("%d\n",GetNext(x));
    }
    return 0;
}