AcWing 838. 堆排序
原题链接
简单
作者:
跟着灿哥学切菜
,
2021-01-15 17:18:26
,
所有人可见
,
阅读 260
#include <iostream>
using namespace std;
const int N = 100010;
int h[N], s;
void down(int u) {
//就是来验证此时的节点是否是最小值(这个节点与其两个孩子节点)
//使用t来表示三个点中最小值的下标
int t = u;
//如果左孩子存在, 并且左孩子的值小于根节点
if (u * 2 <= s && h[u * 2] < h[t]) t = u * 2;
if (u * 2 + 1 <= s && h[u * 2 + 1] < h[t]) t = u * 2 + 1;
if (u != t) {
swap(h[u], h[t]);
down(t);
}
}
int main() {
int n, m;
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i ++) scanf("%d", &h[i]);
s = n;
for (int i = n / 2; i >= 1; i --) down(i);
while (m --) {
printf("%d ", h[1]);
h[1] = h[s];
s --;
down(1);
}
return 0;
}