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

题目描述

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

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

输入 #1复制

?

1 2 3 4 5 6 7 8

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

输出 #1复制

?

1 2 3 4

2 3 1 5

输入里面前面的5 是插入，运用lower_bound、upper_bound

每一个可用 switch语句

#include <iostream>
#include <bits/stdc++.h>
using namespace std;
int main()
{
    int a[10002];
    int q,b,c,cnt,p;
    cin>>q;
    while(q--)
    {
        cin>>b;
        switch(b){
            case 1:
                cin>>c;
                cout<<lower_bound(a+1,a+cnt+1,c)-a<<endl;
                break;
            case 2:
                cin>>c;
                cout<<a[c]<<endl;
                break;
            case 3:
                cin>>c;
                p=lower_bound(a+1,a+cnt+1,c)-a;
                if(p==1) cout<<"-2147483647"<<endl;
                else cout<<a[p-1]<<endl;
                break;
            case 4:
                cin>>c;
                p=upper_bound(a+1,a+cnt+1,c)-a;
                if(p==cnt+1) cout<<"2147483647"<<endl;
                else cout<<a[p]<<endl;
                break;
            case 5:
                cin>>c;
                a[++cnt]=c;
                sort(a+1,a+cnt+1);//排序方便寻找
                break;
        }
    }
    return 0;
}