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

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

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

提示

请用堆排序完成。 

#include <stdio.h>
#include <string.h>

void HeapAdjust(int a[], int s, int t) //筛选
{
    int j;
    int rc=a[s];//根节点（堆顶记录数）
    for (j=2*s; j<=t; j*=2)//从每个节点的左子树开始遍历
    {
        if (j<t&&a[j]>a[j+1]) //比较左右子树取较小值
            ++j;
        if (rc<a[j]) //若当前节点小于其左右子树则无需交换
            break;
        a[s] = a[j];
        s=j;
    }
    a[s] = rc; //将堆顶记录数存到最后一个比较的位置
} // HeapAdjust

void HeapSort (int a[], int n)
{
    int i, j, t;
    for (i=n; i>1; i--) //依次进行筛选
    {
        t = a[1];
        a[1] = a[i];
        a[i] = t;    //将堆顶元素与当前未经排序的序列的最后一个元素进行交换
        HeapAdjust (a, 1, i-1);
    }
}

int main()
{
    int a[20], i, j, n, m;
    scanf("%d %d", &n, &m);
    for(i=1; i<=m; i++)
        scanf("%d", &a[i]);
    for (i=m/2; i>0; i--) //建小顶堆，从最后一个有子树的节点起逐渐向上调整
        HeapAdjust(a, i, m);
    for(i=m+1; i<=n; i++)
    {
        int num;
        scanf("%d", &num);
        if(num>a[1])
        {
            a[1] = num;
            for (j=m/2; j>0; j--) //建小顶堆，从最后一个有子树的节点起逐渐向上调整
                HeapAdjust(a, j, m);
        }


    }

    HeapSort(a, m);
     for(i=1; i<=m; i++)
    {
        if(i==1)
            printf("%d", a[i]);
        else
            printf(" %d", a[i]);
    }

}