重铸华农荣光 我辈义不容辞

the twelfth day - heapsort

#include<iostream>
using namespace std;

const int N=100010;
int n,m;
int h[N],cnt;

void down(int u)
{
    int t=u;
    if(2*u<=cnt&&h[2*u]<h[t]) t=2*u;
    if(2*u+1<=cnt&&h[2*u+1]<h[t]) t=2*u+1;
    if(u!=t)
    {
        swap(h[u],h[t]);
        down(t);
    }
}

int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++) cin>>h[i];
    cnt=n;

    for(int i=n/2;i;i--) down(i);

    while(m--)
    {
        cout<<h[1]<<" ";
        h[1]=h[cnt--];
        down(1);
    }

    return 0;
}