牛客小白月赛8 数列操作

链接：https://www.nowcoder.com/acm/contest/214/F
来源：牛客网
 

题目描述

clccle是个蒟蒻,她经常会在学校机房里刷题,也会被同校的dalao们虐,有一次,她想出了一个毒瘤数据结构,便兴冲冲的把题面打了出来,她觉得自己能5s内切掉就很棒了,结果evildoer过来一看,说:"这思博题不是1s就能切掉嘛",clccle觉得自己的信心得到了打击,你能帮她在1s中切掉这道水题嘛?

你需要写一个毒瘤(划掉)简单的数据结构,满足以下操作
1.插入一个数x(insert)
2.删除一个数x(delete)(如果有多个相同的数,则只删除一个)
3.查询一个数x的排名(若有多个相同的数，就输出最小的排名)
4.查询排名为x的数
5.求一个数x的前驱
6.求一个数x的后继

输入描述:

第一行,输入一个整数n,表示接下来需要输入n行

接下来n行,输入 一个整数num和一个整数x

输出描述:

当num为3,4,5,6时,输出对应的答案

示例1

输入

复制

8
1 10
1 20
1 30
3 20
4 2
2 10
5 25
6 -1

输出

复制

2
20
20
20

代码：

#include<bits/stdc++.h>
using namespace std;
const int MAXN = (int)1e5 + 5;
int n, num, x;
vector<int> a;
int main()
{
    ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
    cin >> n;
    for (int i = 1; i <= n; ++i)
    {
        cin >> num >> x;
        switch (num)
        {
        case 1:
            a.insert(lower_bound(a.begin(), a.end(), x), x);
            break;
        case 2:
            a.erase(lower_bound(a.begin(), a.end(), x));
            break;
        case 3:
            cout << lower_bound(a.begin(), a.end(), x) - a.begin() + 1 << endl;
            break;
        case 4:
            cout << *(a.begin() + x - 1) << endl;
            break;
        case 5:
            cout << *(lower_bound(a.begin(), a.end(), x) - 1) << endl;
            break;
        case 6:
            cout << *upper_bound(a.begin(), a.end(), x) << endl;
            break;
        }
    }
}