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

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

 

 Code:

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1010;
const int minn = -10001;

int a[25];
int n,m;

//堆排序是倒序的
//所以要找前m大的要维护小顶堆

void HeapAdjust(int s,int mm){
    int key = s;
    int l = s*2;
    int r = s*2+1;
    if( s<=mm/2 ){//如果是大于mm/2 就是叶子结点了,就不需要调整了
        //让key指向最有子树中更小的那个
        if( l<=mm && a[key]>a[l] ) key = l;
        if( r<=mm && a[key]>a[r] ) key = r;
        if( key!=s ){
            int t = a[key];
            a[key] = a[s];
            a[s] = t;
            HeapAdjust(key,mm);
        }
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){//注意这里现在只存了m个数,维护了一个m个节点的堆,如果是n个节点然后去调整的话就会TLE
        scanf("%d",&a[i]);
    }
    for(int i=m/2;i>0;i--){
        HeapAdjust(i,m);
    }
    for(int i=m+1;i<=n;i++){
        int x; scanf("%d",&x);
        if( x>a[1] ){//如果说前维护的这m个元素中的最小的那个元素比当前输入的这个小的话,那么就可以直接删除了,因为他肯定不是前m大的了,然后调整小顶堆维护前m个中的最小值
            a[1] = x;
            HeapAdjust(1,m);
        }
    }
    for(int i=m;i>0;i--){//调整,使得从大到小排
        int t = a[1];
        a[1] = a[i];
        a[i] = t;
        HeapAdjust(1,i-1);
    }

    for(int i=1;i<=m;i++){
        if( i==m ) printf("%d\n",a[i]);
        else printf("%d ",a[i]);
    }
    return 0;
}