838. 堆排序
输入一个长度为 𝑛 的整数数列,从小到大输出前 𝑚 小的数。
输入格式
第一行包含整数 𝑛 和 𝑚。
第二行包含 𝑛 个整数,表示整数数列。
输出格式
共一行,包含 𝑚 个整数,表示整数数列中前 𝑚 小的数。
数据范围
输入样例:
5 3
4 5 1 3 2
输出样例:
1 2 3
堆的相关知识
堆是完全二叉树。
小根堆:父节点小于等于左右儿子,根节点为最小值。
存储堆:用一维数组来存储(从1位开始),x的左儿子:2x,x的右儿子:2x+1。
下沉(down)操作:从根节点下沉,换左右结点小的那个。
上浮(up)操作:上浮,只需比较父节点,小就上浮。
堆的五种操作
1.插入一个数。
heap[ ++ size] = x; up(size);
在尾部插入节点并up。
2.求集合当中的最小值。
heap[1];
小根堆,根节点即为最小值。
3.删除最小值。
heap[1] = heap[size]; szie -- ; down(1);
把最后一个元素覆盖根节点元素然后删除最后一个元素根节点再down。
(因为一维数组删头结点难删尾节点容易)
up和down时间复杂度为logn,使用插入删除为logn。
4.删除任意一个元素。
heap[k] = heap[size]; size -- ; down(k); up(k);//down up只会执行一个,省代码。
5.修任意一个元素。
heap[k] = x; down(k); up(k);
堆的建立调整
for (int i = n / 2; i; i -- ) down(i);
o(n)时间建堆。
我们只需依次将以序号为n/2,n/2 - 1,.......1的结点为根的子树均调整为堆即可。
题解代码
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 100010;
int n, m;
int h[N], cnt;
void down(int u)//u是位置编号
{
int t = u;
if (u * 2 <= cnt && h[u * 2] < h[t]) t = u * 2;//左孩子
if (u * 2 + 1 <= cnt && 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]);
cnt = n;
for (int i = n / 2; i; i -- ) down(i);//调整为堆
while (m -- )
{
printf("%d ", h[1]);
h[1] = h[cnt -- ];
down(1);
}
puts("");
return 0;
}