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

Code

---

使用一个简化版的优先队列求解

#include <bits/stdc++.h>
using namespace std;
int k;   // 记录堆的实际容量
void adjustHeap(int a[], int i)
{
    int child = i;
    if(i == 0)
        return;
    int parent = (i-1) / 2;
    if(a[child] > a[parent])
    {
        swap(a[child], a[parent]);
        adjustHeap(a, parent);
    }
}
int heap_pop(int a[])
{
    int temp = a[0];
    a[0] = -1;
    for(int j = k; j > 0; j--)
    {
        adjustHeap(a, j);
    }
    return temp;
}
int main()
{
    int n,M;
    int a[12];
    ios::sync_with_stdio(false);
    cin>>n>>M;
    int key;
    for(int i = 0; i < n; i++)
    {
        k = i<10?i:9;
        cin>>key;
        a[k] = key;
        adjustHeap(a, k);
    }
    for(int i = 0; i < M-1; i++)
    {
        cout<<heap_pop(a)<<" ";
    }
    cout<<heap_pop(a)<<endl;

    return 0;
}