PTA 7-38 寻找大富翁(优先队列的使用)
本题考点:
- 优先队列的使用
胡润研究院的调查显示,截至2017年底,中国个人资产超过1亿元的高净值人群达15万人。假设给出N个人的个人资产值,请快速找出资产排前M位的大富翁。
学会使用优先队列,不仅要会大顶堆(优先队列),也要会小顶堆的使用:
priority_queue<int, vector<int>, greater<int>> pq;
完整代码如下:
/*
寻找大富翁
优先队列
*/
#include <iostream>
#include <queue>
using namespace std;
int main()
{
int N, M;
priority_queue<long> pq;
scanf("%d%d", &N, &M);
long num;
while (N--)
{
scanf("%ld", &num);
pq.push(num);
}
bool flag = false;
for (int i = 0; i < M && i < N; i++)
{
if(flag) {
printf(" %d", pq.top());
pq.pop();
}else {
printf("%d", pq.top());
pq.pop();
flag = true;
}
}
return 0;
}