2015年胡润研究院的调查显示,截至2014年9月,个人资产在600万元以上高净值人群达290万人。假设给出N个人的个人资产值,请快速找出资产排前M位的大富翁。
输入格式:
输入首先给出两个正整数N(≤)和M(≤),其中N为总人数,M为需要找出的大富翁数;接下来一行给出N个人的个人资产值,以百万元为单位,为不超过长整型范围的整数。数字间以空格分隔。
输出格式:
在一行内按非递增顺序输出资产排前M位的大富翁的个人资产值。数字间以空格分隔,但结尾不得有多余空格。
输入样例:
8 3
8 12 7 3 20 9 5 18
输出样例:
20 18 12
#include<bits/stdc++.h>
using namespace std;
void HeapAdjust(int data[],int s,int e)
{
int tmp = data[s];//保存当前结点
int i = 2 * s + 1;//左子树数组位置
while(i<=e)
{
if(i + 1<= e&&data[i + 1] < data[i])
i++;//找到左右子树孩子中最x小的
if(data[i] >= tmp)
break;//符合堆定义,不用调整
data[s] = data[i];
s = i;
i = 2 * s + 1;//最小子节点往上移动,代替父节点
}
data[s] = tmp;
}
void CreatHeap(int data[],int N)
{
for(int i = N/2 - 1; i>0; i --) //将数组建立成最小堆,把第一个位置序号为 N/2 -1
HeapAdjust(data, i, N-1);
}
void HeapSort(int data[],int N)
{
CreatHeap(data,N);
for(int i = N - 1; i>0; i--)
{
int tmp = data[i];
data[i] = data[0];
data[0] = tmp;
HeapAdjust(data,0,i - 1);
}
}
int main()
{
int N,M;
cin>>N>>M;
if(M >N)
{
M = N;
}
int data[N];
for(int i = 0; i<N; i++)
{
scanf("%d",&data[i]);//时间限制!!!
}
HeapSort(data,N);
cout<<data[0];
for(int i = 1;i<M;i++)
{
cout<<" "<<data[i];
}
return 0;
}