7-1 寻找大富翁 PTA 堆排序

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

简单的排序问题，快排超时了，换了堆排序就过了

AC代码：

#include <iostream>

#include <cstring>

#include <algorithm>

#include <queue>

#include <vector>

#include <cstdio>

#include <malloc.h>

#define INF 0x3f3f3f3f

#define FRER() freopen("in.txt", "r", stdin)

#define FREW() freopen("out.txt", "w", stdout)

using namespace std;

const int maxn = 1e6 + ;

int num[maxn];

void adjust(int s, int m) {

    int temp = num[s];

    for(int j = s * ; j <= m; j *= ) {

        if(j < m && num[j] > num[j + ]) ++j;

        if(temp <= num[j]) break;

        num[s] = num[j];

        s = j;

    }

    num[s] = temp;

}

void heapSort(int n) {

    for(int i = n / ; i > ; --i)

        adjust(i, n);

    for(int i = n; i > ; --i) {

        swap(num[], num[i]);

        adjust(, i - );

    }

}

int main()

{

    int n, m;

    scanf("%d %d", &n, &m);

    for(int i = ; i <= n; ++i)

        scanf("%d", &num[i]);

    heapSort(n);

    for(int i = ; i <= m && i <= n; ++i) {

        printf("%d%c", num[i], (i == m || i == n) ? '\n' : ' ');

    }

    return ;

}