第 k 小牛客小白月赛30

最新推荐文章于 2022-09-29 20:24:26 发布

这么着还是那么着

最新推荐文章于 2022-09-29 20:24:26 发布

阅读量291

点赞数

原文链接：https://blog.nowcoder.net/n/c4530c23649f4672ac4e8a48dace9563

版权

第 k 小

题目描述

有一个长度为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 小的数。

不太会写，转一下别人的代码学一下
优先队列+模拟

#include <queue>
#include <cmath>
#include <cstdio>
#include <string>
#include <iostream>
#include <algorithm>
 
using namespace std;
#define endl "\n"
typedef long long ll;
typedef unsigned long long ull;
const ll mod = 1e9+7;
priority_queue<int> que;
const int maxn = 2e5+55;
int a[maxn];
int main() {
    int n, m, k;
    cin >> n >> m >> k;
    for (int i = 0; i < n; ++ i) {
        cin >> a[i];
    }
    sort(a, a+n);
    for (int i = 0; i < n && i < k; ++ i) {
        que.push(a[i]);
    }
    int cnt = min(n, k);
 
    while (m --) {
        int x, y;
        cin >> x;
        if (x == 1) {
            cin >> y;https://blog.nowcoder.net/n/c4530c23649f4672ac4e8a48dace9563
            cnt ++;
            que.push(y);
            if (cnt > k) {
                cnt --;
                que.pop();
            }
        }
        else {
            if (cnt == k) {
                cout << que.top() << endl;
            }
            else cout << -1 << endl;
        }
    }
    return 0;
}