AcWing 838. 堆排序

 

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