[AcWing 838] 堆排序

#include<iostream>

using namespace std;
const int N = 1e5 + 10;
int h[N], sz;
void down(int u)
{
    int t = u;
    if (2 * u <= sz && h[2 * u] < h[t])     t = 2 * u;
    if (2 * u + 1 <= sz && h[2 * u + 1] < h[t])     t = 2 * u + 1;
    if (u != t) {
        swap(h[t], h[u]);
        down(t);
    }
}
int main()
{
    int n, m;
    cin >> n >> m;
    for (int i = 1; i <= n; i ++)   cin >> h[i];
    sz = n;
    for (int i = sz / 2; i; i --)   down(i);
    while (m --) {
        cout << h[1] << ' ';
        h[1] = h[sz];
        sz --;
        down(1);
    }
    return 0;
}