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 <iostream>
#include<bits/stdc++.h>
using namespace std;
int heap[11];
void downAdjust(int low,int high)//向下调整
{
int i=low,j=i*2;
while(j<=high)
{
if(j+1<=high&&heap[j]>heap[j+1])
j++;
if(heap[i]>heap[j])
{
swap(heap[i],heap[j]);
i=j;
j=i*2;
}
else break;
}
}
void creatHeap(int n)//建堆
{
for(int i=n/2; i>0; i--)
downAdjust(i,n);
}
int main()
{
int n,m,i,x;
scanf("%d %d",&n,&m);
for(i=1; i<=m; i++)
{
scanf("%d",&heap[i]);
}
creatHeap(m);//建一个规模为m的堆(一个小顶堆)
for(i=m+1; i<=n; i++)//输入后面的数字,并依次与堆顶(堆中的最小的元素)比较。
//如果大于堆顶元素,就将它纳入堆中,并对堆进行调整。
{
scanf("%d",&x);
if(x>heap[1])
{
heap[1]=x;
downAdjust(1,m);
}
}
for(i=m; i>1; i--)//这其实相当于一个堆排序。
{
swap(heap[1],heap[i]);
downAdjust(1,i-1);
}
for(i=1; i<m; i++)printf("%d ",heap[i]);
printf("%d\n",heap[m]);
return 0;
}
调整的过程也可以这样写:
#include <iostream>
#include<bits/stdc++.h>
using namespace std;
int heap[11];
void downAdjust(int low,int high)//调整
{
int j;
int rc=heap[low];
for(j=2*low; j<=high; j=j*2)
{
if(j+1<=high&&heap[j]>heap[j+1])
j++;
if(rc<=heap[j])break;
heap[low]=heap[j];
low=j;
}
heap[low]=rc;
}
void creatHeap(int n)//建堆
{
for(int i=n/2; i>0; i--)
downAdjust(i,n);
}
int main()
{
int n,m,i,x;
scanf("%d %d",&n,&m);
for(i=1; i<=m; i++)
{
scanf("%d",&heap[i]);
}
creatHeap(m);//建一个规模为m的堆(一个小顶堆)
for(i=m+1; i<=n; i++)//输入后面的数字,并依次与堆顶(堆中的最小的元素)比较。
//如果大于堆顶元素,就将它纳入堆中,并对堆进行调整。
{
scanf("%d",&x);
if(x>heap[1])
{
heap[1]=x;
downAdjust(1,m);
}
}
for(i=m; i>1; i--)//这其实相当于一个堆排序。
{
swap(heap[1],heap[i]);
downAdjust(1,i-1);
}
for(i=1; i<m; i++)printf("%d ",heap[i]);
printf("%d\n",heap[m]);
return 0;
}