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;
}