01trie(平衡树)

 

tyvj 普通平衡树为例。

 

题目描述

这是一道模板题。

您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作：

1. 插入 x xx 数；
2. 删除 x xx 数（若有多个相同的数，因只删除一个）；
3. 查询 x xx 数的排名（若有多个相同的数，因输出最小的排名）；
4. 查询排名为 x xx 的数；
5. 求 x xx 的前趋（前趋定义为小于 x xx，且最大的数）；
6. 求 x xx 的后继（后继定义为大于 x xx，且最小的数）。

输入格式

第一行为 n nn，表示操作的个数，下面 n nn 行每行有两个数 opt \mathrm{opt}opt 和 x xx，opt \mathrm{opt}opt 表示操作的序号（1≤opt≤6 1 \leq \mathrm{opt} \leq 61≤opt≤6）。

输出格式

对于操作 3、4、5、6 每行输出一个数，表示对应答案。

 

 

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cstdlib>

using namespace std;

const int N = 110000 * 35;

int ch[N][2], sum[N], tot = 1;

#define walk for(int i = 31, t, rt = 1 ; (t = (val >> i) & 1), ~i ; i --)

void ins(int val, int c = 1) {
    walk {
        if(ch[rt][t] == 0) ch[rt][t] = ++ tot;
        sum[rt = ch[rt][t]] += c;
    }
}

int kth(int val, int ret = 0) {
    walk {
        if(val > sum[ch[rt][0]]) {
            val -= sum[ch[rt][0]];
            ret |= 1 << i;
            rt = ch[rt][1];
        } else {
            rt = ch[rt][0];
        }
    }
    return ret;
}

int rk(int val, int ret = 0) {
    walk {
        ret += t * sum[ch[rt][0]];
        rt = ch[rt][t];
    }
    return ret;
}

int pre(int val) {
    return kth(rk(val));
}

int sub(int val) {
    return kth(rk(val + 1) + 1);
}

int n, opt, x;

int main() {
    scanf("%d", &n);
    for(int i = 1 ; i <= n ; i ++) {
        scanf("%d%d", &opt, &x);
        x += int(1e7);
        if(opt == 1) ins(x);
        if(opt == 2) ins(x, -1);
        if(opt == 3) printf("%d\n", rk(x) + 1);
        if(opt == 4) printf("%d\n", kth(x - (int)1e7) - (int)1e7);
        if(opt == 5) printf("%d\n", pre(x) - (int)1e7);
        if(opt == 6) printf("%d\n", sub(x) - (int)1e7);
    }
}