838. 堆排序

AC代码：

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

const int N=1e5+10;

int n,m;
int h[N],cnt;
//堆的下沉操作
void down(int k){
    //找到k和两个子结点中的最小值
    int t=k;
    if(k*2<=cnt&&h[t]>h[k*2]) t=k*2;
    if(k*2+1<=cnt&&h[t]>h[k*2+1]) t=k*2+1;
    //如果k不是最小值
    if(t!=k){
        swap(h[t],h[k]);
        down(t);
    }
}
int main(){
    cin>>n>>m;
    cnt=n;
    for(int i=1;i<=n;i++) cin>>h[i];//初始化
    for(int i=n/2;i;i--) down(i);//构建堆，从最后一个值的父节点开始down
    while(m--){
        cout<<h[1]<<" ";//输出最小值
        h[1]=h[cnt];
        cnt--;
        down(1);
    }
    return 0;
}