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

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

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<stdio.h>
void swap(int *c, int *b)
{
    int t = *c;
    *c = *b;
    *b = t;
}
void down(int a[], int n, int i)//间最小堆的down操作
{
    while((i = 2 * i) <= n)
    {
        if(i + 1 <= n && a[i + 1] < a[i])swap(a + i + 1, a + i);
        if(a[i] < a[i / 2])swap(a + i, a + i / 2);
        else break;
    }
}
void makeheap(int a[], int n)//建最小堆
{
    for(int i = n / 2; i > 0; i--)
    down(a, n, i);
}
void heapsort(int a[], int m)//堆排序（该堆排序后数组为降序）
{
    makeheap(a, m);
    for(int i = m; i >= 1; i--)
    {
        swap(a + i, a + 1);
        down(a, i - 1, 1);
    }
}
int main()
{
    int n, m;
    scanf("%d%d", &n, &m);
    int i, a[20];
    for(i = 1; i <= m; i++)
    {
        scanf("%d", &a[i]);
    }
    makeheap(a, m);//建m大小的堆
    for(i = m + 1; i <= n; i++)//m后的数
    {
        int x;
        scanf("%d", &x);
        if(x > a[1])//判断大小 大于堆顶数的执行down操作
        {
            a[1] = x;
            down(a, m, 1);
        }
    }
    heapsort(a, m);//堆排序
    for(i = 1; i <= m; i++)//输出
    {
        if(i == 1)printf("%d", a[i]);
        else printf(" %d", a[i]);
    }
    printf("\n");
    return 0;
}