BZOJ 3224 普通平衡树

Description
您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作：
1. 插入x数
2. 删除x数(若有多个相同的数，因只删除一个)
3. 查询x数的排名(若有多个相同的数，因输出最小的排名)
4. 查询排名为x的数
5. 求x的前驱(前驱定义为小于x，且最大的数)
6. 求x的后继(后继定义为大于x，且最小的数)

---

【题目分析】
Treap复习，随机种子是我的生日（- -b）

---

【代码】

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>

#include <set>
#include <map>
#include <string>
#include <algorithm>
#include <vector>
#include <iostream>
#include <queue>
using namespace std;

#define maxn 100005

int read()
{
    int x=0,f=1; char ch=getchar();
    while (ch<'0'||ch>'9') {if (ch=='-') f=-1; ch=getchar();}
    while (ch>='0'&&ch<='9') {x=x*10+ch-'0'; ch=getchar();}
    return x*f;
}
int tot,rt,siz[maxn],lc[maxn],rc[maxn],num[maxn],tim[maxn],val[maxn];

void update (int k)
{
    siz[k]=siz[lc[k]]+siz[rc[k]]+tim[k];
}

void lturn (int &k)
{
    int t=rc[k];
    rc[k]=lc[t];
    lc[t]=k;
    siz[t]=siz[k];
    update(k);
    k=t;
    return ;
}

void rturn (int &k)
{
    int t=lc[k];
    lc[k]=rc[t];
    rc[t]=k;
    siz[t]=siz[k];
    update(k);
    k=t;
    return ;
}

void ins (int &k,int x)
{
    if (!k)
    {
        k=++tot;
        siz[k]=tim[k]=1;
        num[k]=x;
        val[k]=rand();
        return ;
    }
    siz[k]++;
    if (x==num[k])
    {
        tim[k]++;
        return ;
    }
    if (x<num[k])
    {
        ins(lc[k],x);
        if (val[lc[k]]>val[k]) rturn(k);
        return ;
    }
    else
    {
        ins(rc[k],x);
        if (val[rc[k]]>val[k]) lturn(k);
        return ;
    }
}
void del (int &k,int x)
{
    if (num[k]==x)
    {
        if (tim[k]>1)
        {
            tim[k]--;
            siz[k]--;
            return ;
        }
        else if (lc[k]*rc[k]==0)
        {
            k=lc[k]+rc[k];
            return ;
        }
        else if (val[lc[k]]>val[rc[k]])
        {
            rturn(k);
            del(k,x);
            return ;
        }
        else
        {
            lturn (k);
            del(k,x);
            return ;
        }
    }
    siz[k]--;
    if (x<num[k]) del(lc[k],x);
    else del (rc[k],x);
}

int find (int k,int x)
{
    if (siz[lc[k]]<x&&siz[lc[k]]+tim[k]>=x) return num[k];
    else if (siz[lc[k]]>=x) return find(lc[k],x);
    else return find(rc[k],x-siz[lc[k]]-tim[k]);
}

int rank (int k,int x)
{
    if (!k) return -0x3f3f3f3f;
    if (num[k]==x) return siz[lc[k]];
    else if (x<num[k]) return rank(lc[k],x);
    else return siz[lc[k]]+tim[k]+rank(rc[k],x);
}

int ans;

void pre (int k,int x)
{
    if (!k) return ;
    if (num[k]>=x) return pre(lc[k],x);
    ans=num[k]; return pre(rc[k],x);
}

void nxt (int k,int x)
{
    if (!k) return ;
    if (num[k]<=x) return nxt(rc[k],x);
    ans=num[k]; return nxt(lc[k],x);
}

int main()
{
    srand(20000416);
    int m=read();
    while (m--)
    {
        int opt=read(),x=read();
        switch (opt)
        {
            case 1: ins(rt,x); break;
            case 2: del(rt,x); break;
            case 3: printf("%d\n",rank(rt,x)+1); break;
            case 4: printf("%d\n",find(rt,x)); break;
            case 5: pre(rt,x); printf("%d\n",ans); break;
            case 6: nxt(rt,x); printf("%d\n",ans);
        }
    }
    return 0;
}