7-1 寻找大富翁(PTA - 数据结构)

胡润研究院的调查显示，截至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

---

提交结果： 

---

思路分析：

        乍一看输出前几个最大的，一脑子都是堆排序，结果排序完了，发现打印排好序的前几个元素也行。

        要注意的时，m和n的大小问题，整道题难度倒也不大。 

---

代码：

//
// Created by DDD on 2023/12/22.
//
#include 

void Swap(int* pa, int* pb)
{
    int tmp = *pa;
    *pa = *pb;
    *pb = tmp;
}

void AdjustDown(int* arr, int n, int root)
{
    int parent = root;
    int child = parent * 2 + 1;
    while (child < n)
    {
        if (child + 1 < n && arr[child + 1] < arr[child])
        {
            child = child + 1;
        }
        if (arr[child] < arr[parent])
        {
            Swap(&arr[child], &arr[parent]);
            parent = child;
            child = parent * 2 + 1;
        }
        else
        {
            break;
        }
    }
}

void PrintArray(int* arr, int n)
{
    int i;
    for (i = 0; i < n; i++)
    {
        if(i == n-1)
            printf("%d", arr[i]);
        else printf("%d ", arr[i]);
    }
}


void HeapSort(int* arr, int n)
{
    int i = 0;
    for (i = (n - 1 - 1) / 2; i >= 0; i--){
        AdjustDown(arr, n, i);
    }
    int end = n - 1;
    while (end > 0){
        Swap(&arr[0], &arr[end]);
        AdjustDown(arr, end, 0);
        --end;
    }
}

int main(){
    int n ,m;
    scanf("%d %d",&n,&m);
    int rich[n];

    for (int i = 0; i < n; ++i) {
        scanf("%d",&rich[i]);
    }
    HeapSort(rich, n);
    if(n