OJ3401数据结构实验之排序四：寻找大富翁

数据结构实验之排序四：寻找大富翁

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