标签:tmp pq int NC214362 个数 cin op
题目
题目描述
有一个长度为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
题解
知识点:优先队列。
这和输出一个固定序列的第 \(k\) 小不同,这里要维护一个动态变化的序列,故用小顶堆维护前 \(k\) 个数,那么队头就是第 \(k\) 个数。
要注意有可能不足 \(k\) 个数。
时间复杂度 \(O(n\log n + m\log m)\)
空间复杂度 \(O(n + m)\)
代码
#include <bits/stdc++.h>
using namespace std;
int main() {
std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
int n, m, k;
cin >> n >> m >> k;
priority_queue<int> pq;
for (int i = 0, tmp;i < n;i++) {
cin >> tmp;
pq.push(tmp);
if (pq.size() > k) pq.pop();
}
while (m--) {
int op;
cin >> op;
if (op == 1) {
int tmp;
cin >> tmp;
pq.push(tmp);
if (pq.size() > k) pq.pop();
}
else if (op == 2) {
cout << (pq.size() == k ? pq.top() : -1) << '\n';
}
}
return 0;
}
标签:tmp,pq,int,NC214362,个数,cin,op 来源: https://www.cnblogs.com/BlankYang/p/16459954.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。