牛客网 2018年东北农业大学春季校赛 L题 wyh的天鹅

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

时间限制：C/C++ 3秒，其他语言6秒
空间限制：C/C++ 262144K，其他语言524288K
64bit IO Format: %lld

题目描述

你们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

思路：

一开始我在网上找了一些关于求第k小的算法，感觉比较难以理解，又考虑了用堆来做，可是也比较麻烦（没办法，我就是懒）。不过后来发现，这题可以直接使用stl来解，我们需要用到vector，其实这个里面没有任何算法的成分在里面，我们还是先看代码：

#include<bits/stdc++.h>
using namespace std;
int main()
{
    int T;
    scanf("%d",&T);
    int n,k,x;
    while(T--){
        vector<int>a;
        scanf("%d%d",&n,&k);
        for(int i=0;i<n;i++){
            scanf("%d",&x);
            a.push_back(x);
        }
        sort(a.begin(),a.end());
        char s[15];
        for(int i=0;i<k;i++){
            scanf("%s %d",s,&x);
            if(s[0]=='q'){cout<<a[a.size()-x]<<endl;}
            else if(s[0]=='i'){a.insert(lower_bound(a.begin(),a.end(),x),x);}
            else if(s[0]=='d'){a.erase(lower_bound(a.begin(),a.end(),x));}
        }
    }
}

这里面唯一让人难以理解的就是插入和擦除了，说一下这个lower_bound,它里面有三个参数，其实标准的有四个，第一个表示要查询的起始位置，第二个表示终点，第三个表示要查询的数，整个函数返回第三个参数所在位置，对于2，20，30,40这个数组，如果查询15，那么返回的结果是20的位置，也就是大于要查询的数的第一个位置。