2018年东北农业大学春季校赛 L-wyh的天鹅

2018年东北农业大学春季校赛 L-wyh的天鹅

链接：https://www.nowcoder.com/acm/contest/93/L

来源：牛客网

题目描述

你们wyh学长小时候住在河边，因为周围的生态环境非常好，所以经常会有天鹅浮在湖面上，每只天鹅都长得不一样，它们偶尔排成一排，偶尔分散开，偶尔也会去其他河畔，wyh学长为了统计它们的个数，编了一个程序赋予它们一个“萌”值，但是这些天鹅很不听话，一会儿会从别的地方游过来一两只，一会儿又会在统计过程中游走一两只，现在请你帮他完成统计任务。

输入描述:

共有T(T<=10)组数据，每组数据第一行为两个数 N, M (N，M <= 500000)，代表有N只天鹅和M次操作，接下来一行是N个数字，下面M行首先会输入一个字符串S，接着会有三类操作，如果S是“insert”，接着输入一个正整数a，代表插入一只“萌”值为a的天鹅，如果S是“delete”，接着输入一个正整数a，代表删除一只“萌”值为a的天鹅，如果S是“query”，接着输入一个正整数k，代表查询“萌”值第k大的天鹅。

萌值为[1,1000000000]，并且保证一定存在第k大

输出描述:

对应每次询问，输出询问结果。

示例1

输入

1
5 4
6 4 2 9 1
query 2
insert 7
delete 6
query 2

输出

6
7

---

思路

此题可以用线段树，但是略复杂。

也可以直接用vector过，个人感觉比较好理解。

AC代码

#include <iostream>
#include <vector>
#include <algorithm>
typedef long long ll;
using namespace std;

int main() {
    int T;
    cin >> T;
    while(T--) {
        vector<int> vec;
        int N,M;
        cin >> N >> M;
        while(N--) {
            int a;
            cin >> a;
            vec.push_back(a);
        }
        sort(vec.begin(),vec.end());
        while(M--) {
            string s;
            int b;
            cin >> s >> b;
            if(s[0]=='q') {
                cout << vec.at(vec.size()-b) << endl;
            } else if(s[0]=='i') {
                vec.insert(lower_bound(vec.begin(),vec.end(),b),b);
            } else {
                vec.erase(lower_bound(vec.begin(),vec.end(),b));
            }
        }
    }
    return 0;
}