BZOJ3224 Tyvj 1728 普通平衡树

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

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 5;
int tot = 1;
int op[maxn], a[maxn], b[maxn];
struct node{ int l, r, sum; } tree[maxn*4];
inline int read()
{
    int s = 0, f = 1; char ch = getchar();
    while(ch < '0' || ch > '9'){ if(ch == '-') f = -1; ch = getchar(); }
    while(ch >= '0' && ch <= '9'){ (s *= 10) += ch-'0'; ch = getchar(); }
    return s*f;
}
void build(int k,int l,int r)
{
    tree[k].l = l; tree[k].r = r; tree[k].sum = 0;
    if(l == r) return ;
    int mid = (l+r)>>1;
    build(k<<1,l,mid); build(k<<1|1,mid+1,r);
}
void update(int k,int p,int x)
{
    tree[k].sum += x;
    if(tree[k].l == tree[k].r) return ;
    int mid = (tree[k].l+tree[k].r)>>1;
    if(mid >= p) update(k<<1,p,x);
    else update(k<<1|1,p,x);
}
int Rank(int k,int num)
{
    if(tree[k].l == tree[k].r) return 1;
    int mid = (tree[k].l+tree[k].r)>>1;
    if(mid >= num) return Rank(k<<1,num);
    else return Rank(k<<1|1,num)+tree[k<<1].sum;
}
int Kth(int k,int num)
{
    if(tree[k].l == tree[k].r) return tree[k].l;
    int d = tree[k<<1].sum;
    if(d >= num) return Kth(k<<1,num);
    else return Kth(k<<1|1,num-d);
}
int main()
{
    int n, cnt = 0;
    n = read();
    for(int i = 1; i <= n; i++){
        op[i] = read(); a[i] = read();
        if(op[i] != 4) b[++cnt] = a[i];
    }
    sort(b+1,b+cnt+1);
    cnt = unique(b+1,b+cnt+1)-b-1;
    for(int i = 1; i <= n; i++){
        if(op[i] != 4) a[i] = lower_bound(b+1,b+cnt+1,a[i])-b;
    }
    build(1,1,cnt);
    for(int i = 1; i <= n; i++){
        switch(op[i]){
            case 1: update(1,a[i],1); break;
            case 2: update(1,a[i],-1); break;
            case 3: printf("%d\n",Rank(1,a[i])); break;
            case 4: printf("%d\n",b[Kth(1,a[i])]); break;
            case 5: printf("%d\n",b[Kth(1,Rank(1,a[i])-1)]); break;
            default: printf("%d\n",b[Kth(1,Rank(1,a[i]+1))]);
        }
    }
    return 0;
}