P5076 【深基16.例7】普通二叉树（简化版）

P5076 【深基16.例7】普通二叉树（简化版）

题目描述

您需要写一种数据结构，来维护一些数（ 都是 10^9109 以内的数字）的集合，最开始时集合是空的。其中需要提供以下操作，操作次数 qq 不超过 10^4104：

1. 查询 xx 数的排名（排名定义为比当前数小的数的个数 +1+1。若有多个相同的数，因输出最小的排名）。
2. 查询排名为 xx 的数。
3. 求 xx 的前驱（前驱定义为小于 xx，且最大的数）。若未找到则输出 -2147483647−2147483647。
4. 求 xx 的后继（后继定义为大于 xx，且最小的数）。若未找到则输出 21474836472147483647。
5. 插入一个数 xx。

输入格式

无

输出格式

无

输入输出样例

输入 1

7
5 1
5 3
5 5
1 3
2 2
3 3
4 3

输出 1

2
3
1
5

二叉查找问题可以使用multiset，multiset满足可以重复的集合，但是是有序的，内部查找满足红黑树。lower_bound函数可以找出集合小于等于b的数，upper_bound函数可以找出集合中大于b的数

#include <set>
#include <cstdio>
#include <iostream>
using namespace std;
multiset<int> s;
int N, a, b;
int main() {
    scanf("%d", &N);
    s.insert(-2147483647);
    s.insert(2147483647);
    while(N--) {
        scanf("%d%d", &a, &b);
        if(a == 1) {
            int cnt = 0;
            auto it = s.lower_bound(b);
            for(auto i = s.begin(); i != it; i++, cnt++);
            printf("%d\n", cnt);
        } else if(a == 2) {
            auto it = s.begin();
            while(b--) it++;
            printf("%d\n", *it);
        } else if(a == 3) {
            auto it = --s.lower_bound(b); // 小于等于b的值
            printf("%d\n", *it);
        } else if(a == 4) {
            auto it = s.upper_bound(b); // 大于b的值
            printf("%d\n", *it);
        } else if(a == 5) {
            s.insert(b);
        }
    }
    return 0;
}