堆排-寻找大富翁.cpp

/*

  5-1 寻找大富翁   (25分)
2015年胡润研究院的调查显示，截至2014年9月，个人资产在600万元以上高净值人群达290万人。假设给出N个人的个人资产值，请快速找出资产排前M位的大富翁。
输入格式:

输入首先给出两个正整数NN（\le 10^6≤10
?6
?? ）和MM（\le 10≤10），其中NN为总人数，MM为需要找出的大富翁数；接下来一行给出NN个人的个人资产值，以百万元为单位，为不超过长整型范围的整数。数字间以空格分隔。
输出格式:

在一行内按非递增顺序输出资产排前MM位的大富翁的个人资产值。数字间以空格分隔，但结尾不得有多余空格。
输入样例:

8 3
8 12 7 3 20 9 5 18
输出样例:

20 18 12
*/

//堆排序
#include <stdio.h>
#include <stdlib.h>

void HeapAdjust(long long* r,int s,int n){//建立大顶堆的时候要保证堆顶是大的
    //r[s..n]中除了堆顶s外均满足堆定义，调整其为大顶堆（这就相当于一个小规模的问题）
    r[0] = r[s];//r[0]当做哨所暂存值
    for(int j=2*s;j<=n;j*=2){//j是s的左孩子,j+1是右孩子
        if(j<n&&r[j]<r[j+1])
            j++;
        if(r[0]>=r[j])
            break;
        else{//堆顶s比子孩子小，说明要进行堆的调整
            r[s] = r[j];
            s = j;//进行下一次堆顶的递归
        }
    }
    r[s] = r[0];//把完全二叉树最后一个赋值为
}

int main(){

    int n,m;
    scanf("%d",&n);scanf("%d",&m);
    if(m > n)//坑爹啊！这里也是一个点。前几名的个数必须小于大富翁总数
        m = n;
    long long  r[n+1];
    for(int i=1;i<n+1;i++){
        scanf("%lld",&r[i]);
    }
    //建立初始大顶堆
    for(int i=n/2;i>0;i--)
        HeapAdjust(r,i,n);

    //筛选出前m个大的
    for(int i=n;i>n-m;i--){
        int temp = r[1]; r[1] = r[i];r[i] = temp;//交换堆顶元素与最后的一个元素（然后就把他前面的一个看成最后一个，他就不看了）
        printf("%d",r[i]);
        if(i != n-m+1)
            printf(" ");
        HeapAdjust(r,1,i-1);
    }

    return 0;
}