w7-5 怪物的身高（堆排序）

合集 - 数据结构(12)

1.w1-4 求两个正整数的乘积2023-04-192.w2-4 高精度减法2023-04-193.w3-4 后缀表达式2023-04-19

4.w7-5 怪物的身高（堆排序）2023-04-19

5.w4-4 世界杯2023-04-196.w5-4 验证栈序列2023-04-197.w6-3 怪物的身高（快速排序）2023-04-198.w3-2 寄包柜2023-04-199.w4-1 队列安排2023-04-1910.w5-1 序列合并2023-04-1911.T9-1 美国血统2023-05-1312.T9-2 求二叉树中节点间的宽度2023-05-13

收起

#include <iostream>
using namespace std;
int N,m,A[5000005];
//用scanf，printf卡常
template <class T>
void sift(T a[],int k,int m){//换位，假设左右结点已排好序
    int i=k;//根结点
    int j=2*i;//左结点
    T tmp;
    while(j<=m){//当j是最后一个结点时
        if(j<m&&a[j]>a[j+1]) j++;//如果右结点更小，转到右结点，注意越界
        if(a[i]<a[j]) return;//如果a较小，不管
        else{//否则交换父，子结点
            tmp=a[i];
            a[i]=a[j];
            a[j]=tmp;
            i=j;//换到子节点，下一次判断
            j=2*i;
        }

    }

}
template <class T>
void HeapSort(T a[],int n){
    for(int i=n/2;i>=1;i--){
        sift(a,i,n);//从后往前排序，i是起点，n是终点
    }
    T tmp;
    for(int i=1;i<=m+1;++i){//一共只排m次
        tmp=a[1];//交换根和最后结点
        a[1]=a[n-i+1];
        a[n-i+1]=tmp;
        sift(a,1,n-i);//排序

    }
    printf("%d",a[n-m]);



}
int main() {
    scanf("%d%d",&N,&m);
    for(int i=1;i<=N;++i){
        scanf("%d",&A[i]);
    }
    HeapSort(A,N);

    return 0;
}