堆就是完全二叉树,且根节点的值一定大于或者小于其他所有节点(取决于大根堆或小根堆),完全二叉树上面的节点都是满的,最后一层的节点从左到右排列。
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1e5 + 10;
int n, m, len;
int heap[N];
void down(int u)
{
int t = u;
//左儿子存在且左儿子小于父节点,则父节点下沉
if (u * 2 <= len && heap[u * 2] < heap[t]) t = 2 * u;
//右儿子存在且右儿子小于父节点,则父节点下沉
if (u * 2 + 1 <= len && heap[u * 2 + 1] < heap[t]) t = 2 * u + 1;
if (u != t)
{
swap(heap[u], heap[t]);
down(t); //继续下沉,沉到不能沉为止
}
}
int main()
{
cin >> n >> m;
//下标必须从1开始,若从0开始的话,u*2还是为0,根本不是左子节点下标。
for (int i = 1; i <= n; i ++ ) cin >> heap[i];
len = n;
//最底部先垫一层,再插入,一边插入一遍排序。最底部一层元素数量为总元素的二分之一
for (int i = n / 2; i; i -- ) down(i);
while (m -- )
{
cout << heap[1] << ' ';
heap[1] = heap[len -- ]; //删除堆顶最小值
down(1);
}
return 0;
}