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

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

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

提示

请用堆排序完成。 

来源

xam

示例程序

#include<bits/stdc++.h>
using namespace std;
void heap(int a[],int i,int size)
{
    int temp;
    if(i<=size/2)
    {
        int left=2*i;
        int right=2*i+1;
        int root=i;
        if(left<=size&&a[root]>a[left])
        {
            temp=a[root];
            a[root]=a[left];
            a[left]=temp;
            root=left;
        }
        if(right<=size&&a[root]>a[right])
        {
            temp=a[root];
            a[root]=a[right];
            a[right]=temp;
            root=right;
        }
        if(root!=i)
            heap(a,root,size);
    }
}
int main()
{
    int i,n,m,x;
    int a[11];
    scanf("%d%d",&n,&m);
    for(int i=1; i<=m; ++i)
        scanf("%d",&a[i]);
    for(int i=m/2; i>=1; --i)
        heap(a,i,m);
    for(int j=m+1; j<=n; j++)
    {
        scanf("%d",&x);
        if(x>a[1])
        {
            a[1]=x;
            heap(a,1,m);
        }
    }
    for(int i=m; i>=1; --i)
    {
        swap(a[1],a[i]);
        heap(a,1,i-1);
    }
    for(i=1; i<m; i++)
        printf("%d ",a[i]);
        printf("%d\n",a[m]);
}
 


//课本上的方法
#include <bits/stdc++.h>
using namespace std;
int a[11];
void heapjust(int s,int m)          //把a[]建成小顶堆
{
    int t=a[s];
    for(int i=2*s; i<=m; i*=2)
    {
        if(i<m&&a[i]>a[i+1])  //记录两个孩子节点较小的位置
            i++;
        if(t<a[i])
            break;
        a[s]=a[i];
        s=i;
    }
    a[s]=t;
}
int main()
{
    int n,m,i,j;
    while(~scanf("%d %d",&n,&m))
    {
        for(i=1; i<=m; i++)  //先建成m个数的小顶堆
            scanf("%d",&a[i]);
        for(i=m/2; i>0; i--)  //从m/2开始比较
            heapjust(i,m);
        for(i=m+1; i<=n; i++)
        {
            int x;
            scanf("%d",&x);
            if(x<=a[1])
                continue;
            a[1]=x;   //更新小顶堆
            for(j=m/2; j>0; j--)
                heapjust(j,m);
        }
        for(i=m;i>1;i--) //最后堆排序
        {
            int t=a[1];
            a[1]=a[i];
            a[i]=t;
            heapjust(1,i-1);
        }
        for(i=1; i<=m; i++)
            printf("%d%c",a[i],i==m?'\n':' ');
    }
}