第十章作业2--交换类排序与选择类排序

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);
    }
}