数据结构实验之排序四:寻找大富翁
Time Limit: 200MS Memory Limit: 512KB
Problem Description
2015胡润全球财富榜调查显示,个人资产在1000万以上的高净值人群达到200万人,假设给出N个人的个人资产值,请你快速找出排前M位的大富翁。
Input
首先输入两个正整数N( N ≤ 10^6)和M(M ≤ 10),其中N为总人数,M为需要找出的大富翁数目,接下来给出N个人的个人资产,以万元为单位,个人资产数字为正整数,数字间以空格分隔。
Output
一行数据,按降序输出资产排前M位的大富翁的个人资产值,数字间以空格分隔,行末不得有多余空格。
Example Input
6 3
12 6 56 23 188 60
Example Output
188 60 56
Hint
请用堆排序完成。
#include <bits/stdc++.h>
using namespace std;
//j结点的父结点下标是(j - 1) / 2;
//j左右子结点下标是 2*j+1, 2*j+2;
void HeapFix(int a[],int i, int n) //调整成小堆
{
int temp, j;
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 creatHeap(int a[], int n) //建堆
{
for(int i = n / 2 - 1; i >= 0; i--)
HeapFix(a,i,n);
}
void Heapsort(int a[], int n) //从大到小排序
{
creatHeap(a,n);
while(n > 0)
{
swap(a[0],a[n-1]);
n--;
HeapFix(a,0,n-1);
}
}
int a[1000100];
int main()
{
int n, m, x;
scanf("%d %d", &n, &m);
for(int i = 0; i < m; i++)
scanf("%d", &a[i]);
creatHeap(a,m);
for(int i = m; i < n; i++)
{
scanf("%d", &x);
if(x < a[0])
continue;
a[0] = x;
creatHeap(a,m);
}
Heapsort(a,m);
for(int i = 0; i < m; i++)
printf("%d%c", a[i], i == m - 1 ? '\n' : ' ');
return 0;
}