7-1 寻找大富翁 (25分)
胡润研究院的调查显示,截至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
仅供参考
#include <bits/stdc++.h>
using namespace std;
void HeapSort(int arr[]);
int arr[1000005];
int N, M;
bool cmp(int a,int b) {
return a > b;
}
int main() {
cin >> N >> M;
if (M > N) M = N;
arr[0] = 0;
for (int i = 0; i < N; i++) {
cin >> arr[i];
}
sort(arr, arr + N, cmp);
for (int i = 0; i < M; i++) {
cout << arr[i];
if (i != M - 1)
cout << " ";
}
return 0;
}
void HeapAdjust(int arr[], int s, int m) {
int rc = arr[s];
for (int i = 2 * s; i <= m; i *= 2) {
if (i < m && arr[i] < arr[i + 1]) ++i;
if (rc >= arr[i]) break;
arr[s] = arr[i];
s = i;
}
arr[s] = rc;
}
void HeapSort(int arr[]) {
for (int i = N / 2; i > 0; --i) {
HeapAdjust(arr, i, N);
}
for (int i = N; i > 1; --i) {
swap(arr[1], arr[i]);
HeapAdjust (arr, 1, i - 1);
}
}