AcWing 838. 堆排序

#include <iostream>
#include <algorithm>
using namespace std;
const int N = ;
int n, m;
int h[N], size;
void down(int u) {
    int t = u;  //u这个点的做儿子是u*2，右儿子是u*2+1  完全二叉树
    if (u *  <= size && h[u * ] < h[t]) t = u * ;  //如果左儿子存在，而左儿子比u小
    if (u *  +  <= size && h[u *  + ] < h[t]) t = u *  + ;//右
    if (u != t) {//判断是不是父亲，如果不是，那么交换
        swap(h[u], h[t]);
        down(t);//递归
    }
}
int main() {
    scanf("%d%d", &n, &m);
    for (int i = ; i <= n; i ++ ) scanf("%d", &h[i]);
    size = n;
    for (int i = n / ; i; i -- ) down(i);//优化
    while (m -- ) {
        printf("%d ", h[]);//输出最小
        h[] = h[size -- ];//末端的数字覆盖最小数字的位置
        down();//再down
    }
    puts("");
    return ;
}