堆——维护一个数据集合
如何手写一个堆?
1、插入一个数 ———— hep[ ++ size] = x; up(size);
2、求集合当中的最小值、———— heap[1];
3、删除最小值 ———— heap[1] = heap[size]; size --; down(1);
4、删除任意一个元素———— heap[k] = heap[size]; size --; down(k); up(k);
5、修改任意一个元素———— heap[k] = x; down[k]; up[k];
小根堆:每个点都是小于等于左右儿子的;
存储:用数组存储堆,1是根节点,x的左儿子:2x;x的右儿子:2x + 1;
down(x):如果把某一个值,变大了,就往下移;
up(x):如果把某一个值,变小了,就往上移。
堆排序
问题描述
输入一个长度为n的整数数列,从小到大输出前m小的数。
输入格式
第一行包含整数n和m。
第二行包含n个整数,表示整数数列。
输出格式
共一行,包含m个整数,表示整数数列中前m小的数。
数据范围
1
<
=
m
<
=
n
<
=
1
0
5
1 <= m <= n <= 10^5
1<=m<=n<=105
1
<
=
数
列
中
元
素
<
=
1
0
9
1 <= 数列中元素 <= 10^9
1<=数列中元素<=109
输入样例
5 3
4 5 1 3 2
输出样例
1 2 3
完整代码实现
#include<iostream>
#include<algorithm>
using namespace std;
const int N = 100010;
int n, m;
int h[N], size1;
void down(int u)
{
int t = u;
if (u * 2 <= size1 && h[u * 2] < h[t]) t = u * 2;
if (u * 2 + 1 <= size1 && 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]);
size1 = n;
for (int i = n / 2; i ; i --) down(i);
while (m --)
{
printf ("%d ", h[1]);
h[1] = h[size1];
size1 --;
down(1);
}
return 0;
}