刚开始学,做个笔记记录一下。用c++
题目描述
有一个长度为n的数组,值为 a[i], 牛牛想找到数组中第 k 小的数。比如 1 2 2 3 4 6 中,第 3 小的数就是2.
牛牛觉得这个游戏太简单了,想加一点难度,现在牛牛有 m 个操作,每个操作有两种类型。
1 x 1 代表操作一,给数组中加一个元素 x 。(0 ≤ x ≤ 1e9)
2 2 代表操作二,查询第 k 小的数。如果没有 k 个数就输出−1
输入描述:
第一行有三个整数,n m k,(1≤n,m,k≤2e5)
第二行包含 n 个整数 a[i] ( 0 ≤ a[i] ≤ 1e9)
接下来m行,每行代表一个操作。具体见题目描述
输出描述:
每次查询输出一个第 k 小的数。
示例1
输入:
5 4 3
1 2 3 4 5
2
1 1
1 3
2
输出:
3
2
思路:
开始先声明一个priority_queue<int> q; 注意这个优先队列默认排序是less,也就是队列里的元素从大到小排序
然后输入n,m,k和队列元素,之后就要判断第k小的元素,因为队列自动帮我们排好序了,所以第k小的元素位置就是队列里的倒数第k位,我们只要把前面的元素都pop()出来,pop()的前提是k不能大于q.size(),否则就输出-1,pop()完后队列里的第一个元素就是第k小的元素了,然后我们用q.top()输出元素就可以
代码如下
#include <bits/stdc++.h>
using namespace std;
priority_queue<int> q;
int main()
{
int n, m, k, x, y;
cin >> n >> m >> k;
while (n--)
{
cin >> x;
q.push(x);
}
while (m--)
{
while (q.size() > k)
q.pop();
cin >> x;
if (x & 1)
cin >> y, q.push(y);
else
{
if (q.size() >= k)
cout << q.top() << '\n';
else
cout << '-1\n';
}
}
return 0;
}