SDUTOJ 3401 - 数据结构实验之排序四：寻找大富翁

Problem Description

2015胡润全球财富榜调查显示，个人资产在1000万以上的高净值人群达到200万人，假设给出N个人的个人资产值，请你快速找出排前M位的大富翁。

Input

首先输入两个正整数N( N ≤ 10^6)和M(M ≤ 10)，其中N为总人数，M为需要找出的大富翁数目，接下来给出N个人的个人资产，以万元为单位，个人资产数字为正整数，数字间以空格分隔。

Output

一行数据，按降序输出资产排前M位的大富翁的个人资产值，数字间以空格分隔，行末不得有多余空格。

Sample Input

6 3
12 6 56 23 188 60

Sample Output

188 60 56

Hint

请用堆排序完成。

Source

xam

#include <stdio.h>

int a[15];

void HeapJust(int s, int m){ // 建小顶堆
    int key = s;
    int lchild = s*2; 
    int rchild = s*2+1;
    if (s <= m/2){
        if (lchild <= m && a[lchild] < a[key]){
            key = lchild;
        }
        if (rchild <= m && a[rchild] < a[key]){
            key = rchild;
        }
        // 用key标记s，lchild,rchild中最小的
        if (key != s){ // 将最小的数放在顶部
            int t = a[key];
            a[key] = a[s];
            a[s] = t;
            HeapJust(key, m); // 往下递归
        }
    }
}

int main() {
    int n, m;
    int temp;
    scanf("%d %d", &n, &m);
    for (int i = 1; i <= m; i++){ // 先输入m个数
        scanf("%d", &a[i]);
    }
    /*
    for (int i = m/2; i > 0; i--){
        HeapJust(i, m);
    }
    替换下面的一行代码，效率会更高
    */
    HeapJust(1, m); // 将m个数调整为小顶堆
    for (int i = m+1; i <= n; i++){ // 输入剩下的数
        scanf("%d", &temp);
        if (temp > a[1]){ // 与堆顶元素比较后，若更小则放堆顶，进行调整
            a[1] = temp;
            HeapJust(1, m);
        }
    }
    for (int i = m; i >= 1; i--){ // 将最小的放在最后面，前面的再调整为最小堆
        int t = a[1];
        a[1] = a[i];
        a[i] = t;
        HeapJust(1, i-1);
    }

    for (int i = 1; i <= m; i++){
        if (i != m){
            printf("%d ", a[i]);
        }else{
            printf("%d\n", a[m]);
        }
    }
    return 0;
}