数据结构实验之排序四:寻找大富翁
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
请用堆排序完成。
用最小堆做,开数组不要开n大小的,要开m大小的,因为题目给的内存太小
代码:
#include<bits/stdc++.h>
using namespace std;
void heapadjust(int *a,int s,int m)//最小堆
{
int rc = a[s];
int j;
for(j = 2*s;j <= m; j*=2)
{
if(j<m&&a[j]>a[j+1]) ++j;
if(rc < a[j]) break;
a[s] = a[j];
s = j;
}
a[s] = rc;
return ;
}
void creatheap(int *a,int n)
{
for(int i = n/2; i > 0; i--)
{
heapadjust(a,i,n);
}
return ;
}
void heapsort(int *a,int n,int m)
{
creatheap(a,m);
int t;
for(int i = m+1 ; i <= n ; i++)
{
scanf("%d",&t);
if(t > a[1])
{
a[1] = t;
heapadjust(a,1,m);
}
}
}
void print(int *a,int *b,int m)
{
for(int i = m; i >= 1 ; --i)
{
int x = a[1];
a[1] = a[i];
a[i] = x;
b[i] = x;
heapadjust(a,1,i-1);
}
}
int main()
{
int n, m;
cin >>n >> m;
int a[m+5];//不能开a[n]数组 会超内存限制
int b[m+5];
for(int i = 1; i<= m ;i++)
{
scanf("%d",&a[i]);
}
heapsort(a,n,m);
print(a,b,m);
for(int i = 1; i <= m; i++)
{
printf("%d",b[i]);
if(i == m) printf("\n");
else
printf(" ");
}
return 0;
}