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

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

Time Limit: 200 ms Memory Limit: 512 KiB

Submit Statistic Discuss

Problem 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

Sample Output

188 60 56

Hint

请用堆排序完成。 

Source

xam 

AC代码：

#include <bits/stdc++.h>
using namespace std;


int a[15];


void heap_adj(int s, int m)
{
    int k = s;
    int l = s*2;
    int r = s*2+1;
    int d;
    if(s <= m/2)
    {
        if(l <= m && a[k] > a[l])
            k = l;
        if(r <= m && a[k] > a[r])
            k = r;
        if(s != k)
        {
            d = a[s];
            a[s] = a[k];
            a[k] = d;
            heap_adj(k, m);
        }
    }
}


int main()
{
    int n, m, i, d, x;
    scanf("%d %d", &n, &m);
    for(i = 1; i <= m; i++)
        scanf("%d", &a[i]);
    for(i = m/2; i > 0; i--)
    {
        heap_adj(i, m);
    }
    for(i = m+1; i <= n; i++)
    {
        scanf("%d", &x);
        if(x > a[1]) 
        {
            a[1] = x;
            heap_adj(1, m);
        }
    }
    for(i = m; i >= 1; i--) 
    {
        d = a[1];
        a[1] = a[i];
        a[i] = d;
        heap_adj(1, i-1);
    }
    printf("%d", a[1]);
    for(i = 2; i <= m; i++)
        printf(" %d", a[i]);
    return 0;
}