胡润研究院的调查显示,截至2017年底,中国个人资产超过1亿元的高净值人群达15万人。假设给出N个人的个人资产值,请快速找出资产排前M位的大富翁。
输入格式:
输入首先给出两个正整数N(≤10^6 )和M(≤10),其中N为总人数,M为需要找出的大富翁数;接下来一行给出N个人的个人资产值,以百万元为单位,为不超过长整型范围的整数。数字间以空格分隔。
输出格式:
在一行内按非递增顺序输出资产排前M位的大富翁的个人资产值。数字间以空格分隔,但结尾不得有多余空格。
输入样例:
8 3
8 12 7 3 20 9 5 18
输出样例:
20 18 12
本题对排序效率要求比较高,STL里的sort排序并不能满足其时间限制。
可能会有N>>M,而输出只有M项,所以我们可以仅保留最大的前M项,想用优先队列输出为最大值的逆序,所以用栈将其翻转。想过用以deque为容器的优先队列可能是我不会用,没能实现。
#include <iostream>
#include <queue>
#include <stack>
using namespace std;
int main() {
int N, M, a;
cin >> N >> M;
M = min(N, M);
priority_queue<int, vector<int>, greater<int>>p;
stack<int>s;
for (int i = 0; i < N; ++i) {
cin >> a;
p.push(a);
if (p.size() > M)
p.pop();
}
while (!p.empty()) {
s.push(p.top());
p.pop();
}
for (int i = 0; i < M; ++i) {
if (i)cout << " ";
cout << s.top();
s.pop();
}
return 0;
}