数据结构实验之排序四:寻找大富翁
Time Limit: 150ms Memory limit: 512K 有疑问?点这里^_^
题目描述
2015胡润全球财富榜调查显示,个人资产在1000万以上的高净值人群达到200万人,假设给出N个人的个人资产值,请你快速找出排前M位的大富翁。
输入
首先输入两个正整数N( N ≤ 10^6)和M(M ≤ 10),其中N为总人数,M为需要找出的大富翁数目,接下来给出N个人的个人资产,以万元为单位,个人资产数字为正整数,数字间以空格分隔。
输出
一行数据,按降序输出资产排前M位的大富翁的个人资产值,数字间以空格分隔,行末不得有多余空格。
示例输入
6 3 12 6 56 23 188 60
示例输出
188 60 56
提示
请用堆排序完成。
#include<iostream>
#include<stdio.h>
using namespace std;
int a[20];
void MinHeapAdjust(int *a,int i,int n)//对堆的调整
{
int j,temp;
temp=a[i];
j=2*i+1;
while (j<n)
{
if (j+1<n&&a[j+1]<a[j])//在左右孩子中找最小的
j++;
if (a[j]>=temp) //当找出的最小的值仍大于父节点是终止循环
break;
a[i]=a[j];
i=j;
j=2*i+1;
}
a[i]=temp;
}
void MakeMinHeap(int *a,int n)//建立小顶堆
{
for (int i=n/2-1;i>=0;i--)
{
MinHeapAdjust(a,i,n);
}
}
void MinHeapSort(int *a,int n)//堆排序
{
MakeMinHeap(a,n);
for (int i=n-1;i>=1;i--)
{
swap(a[i],a[0]);
MinHeapAdjust(a,0,i);
}
}
int main()
{
int n,m,x;
scanf("%d%d",&n,&m);
for (int i=0;i<m;i++)
{
scanf("%d",&a[i]);
}
MakeMinHeap(a,m);//建立小顶堆,a[0]为最小值
for (int i=m;i<n;i++)
{
scanf("%d",&x);
if (x<a[0])
continue;
a[0]=x;//当输入的值大于a[0]是替换a[0]并重新创建小顶堆,使a[0]保持为a[0]~a[m-1]中的最小值,否则忽略
MakeMinHeap(a,m);
}
MinHeapSort(a,m);//对a[0]~a[m-1]中数据进行堆排序
for (int i=0;i<m;i++)
{
printf("%d%c",a[i],i==m-1?'\n':' ');
}
return 0;
}