数据结构实验之排序四:寻找大富翁
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
提示
请用堆排序完成。
来源
xam
示例程序
如果要由小到大排序,建立大项堆,最大的需要在最后,大项堆堆顶是最大的,所以1与n交换,最大的就在最后,然后n--,
重复这样操作(也就是HeaoSort这个函数),这样就是排好序的了;由大到小排序也就是这个意思。
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#define M 1000005
int a[M],n;
void swap(int x,int y)
{
int t;
t=a[x];
a[x]=a[y];
a[y]=t;
return ;
}
void siftdown(int k,int i)//向下调整
{
int t,flag=0;//flag标记是否继续调整
while(i*2<=k&&flag==0)
{
if(a[i]>a[i*2])
t=i*2;
else t=i;
if(i*2+1<=k)
{
if(a[t]>a[i*2+1])
t=i*2+1;
}
if(t!=i)
{
swap(t,i);
i=t;//继续向下调整
}
else flag=1;//说明父节点比两个子节点都小,不需要调整了
}
return ;
}
void creat(int m)//建堆
{
int i;
for(i=m/2; i>=1; i--)
siftdown(m,i);
return ;
}
void HeapSort(int k)
{
while(k>1)
{
swap(1,k);
k--;
siftdown(k,1);
}
return ;
}
int main()
{
int m,i,j,k,x;
scanf("%d%d",&n,&m);
for(i=1; i<=m; i++)
scanf("%d",&a[i]);
creat(m);
for(i=m+1; i<=n; i++)
{
scanf("%d",&x);
if(x>a[1])//更新小项堆
{
a[1]=x;
for(j=m/2; j>=1; j--)
siftdown(m,j);
}
}
HeapSort(m);//堆排序
for(i=1; i<m; i++)
printf("%d ",a[i]);
printf("%d\n",a[i]);
}