SDUT D - 数据结构实验之排序四：寻找大富翁(堆排序)

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

Output

188 60 56

Hint

请用堆排序完成。

#include <bits/stdc++.h>
using namespace std;
int a[1000010];
int n,m;
void adjust(int x,int m)
{
    int k=x,l=k*2,r=k*2+1;
    if(x<=m/2)
    {
        if(l<=m&&a[k]>a[l])
            k=l;
        if(r<=m&&a[k]>a[r])
            k=r;
        if(x!=k)
        {
            int t=a[x];
            a[x]=a[k];
            a[k]=t;
            adjust(k,m);
        }
    }
}
int main()
{
    scanf("%d %d",&n,&m);
    for(int i=1; i<=m; i++)
    {
        scanf("%d",&a[i]);
    }
    for(int i=m/2; i>=1; i--)
    {
        adjust(i,m);
    }
    for(int i=m+1; i<=n; i++)
    {
        int x;
        scanf("%d",&x);
        if(x>a[1])
        {
            a[1]=x;
            adjust(1,m);
        }
    }
    for(int i=m; i>=1; i--)
    {
        int t=a[1];
        a[1]=a[i];
        a[i]=t;
        adjust(1,i-1);
    }
    for(int i=1; i<m; i++)
    {
        cout<<a[i]<<" ";
    }
    cout<<a[m];
    return 0;
}