PTA 7-38 寻找大富翁 (25 point(s))

 胡润研究院的调查显示，截至2017年底，中国个人资产超过1亿元的高净值人群达15万人。假设给出N个人的个人资产值，请快速找出资产排前M位的大富翁。

输入格式:

输入首先给出两个正整数N（≤106）和M（≤10），其中N为总人数，M为需要找出的大富翁数；接下来一行给出N个人的个人资产值，以百万元为单位，为不超过长整型范围的整数。数字间以空格分隔。

输出格式:

在一行内按非递增顺序输出资产排前M位的大富翁的个人资产值。数字间以空格分隔，但结尾不得有多余空格。

输入样例:

8 3
8 12 7 3 20 9 5 18

结尾无空行

输出样例:

20 18 12

结尾无空行

// 比第二个代码少了一行判断
#include <bits/stdc++.h>
using namespace std;
int main() {
	int N, M, money;
	cin >> N >> M;
    // 防止N<M情况
	M = min(N, M);
    priority_queue<int, vector<int>, greater<int>> q;
    stack<int> s;

    while(N--){
        cin >> money;
        q.push(money);
    }
    // 压入栈逆序
    while(!q.empty()){
        s.push(q.top());
        q.pop();
    }
    while(M--){
        cout << s.top() << (M == 0 ? "" : " ");
        s.pop();
    }
}

#include <bits/stdc++.h>
using namespace std;
int main() {
	int N, M, money, space = 0;
	cin >> N >> M;
    // 防止N<M情况
	M = min(N, M);
	priority_queue<int, vector<int>, greater<int>>q;
	stack<int>s;
	while(N--) {
		cin >> money;
		q.push(money);

        // 当队列容量超过M则弹出较小元素
        // 少了这个判断也可以 如果新输入的数据是排名前面的必然会被放到前面
		if (q.size() > M)
			q.pop();
	}
    // 压入栈逆序
    while(!q.empty()){
        s.push(q.top());
        q.pop();
    }
	while(M--) {
		if (space++)cout << " ";
		cout << s.top();
		s.pop();
	}
}

 

// 超时且错误代码 内部容器排序不行 且有相同输入
#include <bits/stdc++.h>
using namespace std;
main(){
    set<int> rich;
    int n, m, money;
    cin >> n >> m;
    while(cin >> money){
        rich.insert(money);
    }
    for(auto ppl = rich.rbegin(); m-- != 0 && ppl != rich.rend(); ppl++){
        cout << (ppl == rich.rbegin() ? "" : " ") << *ppl;
    }
}

参考代码