堆排序实现

堆排序

堆也可以称为优先队列（在stl中优先队列就是堆），用完全二叉树的数据结构来实现。
小根堆（一般用于降序）：每个节点的值都小于等于它的左右节点的值

大根堆（一般用于升序）：每个节点的值都大于或等于它的左右子节点的值
堆的存储：用一维数组来存（下面的数字为数组下标，下标从1开始比较好）

1	2	3

1号点是根节点，
x的左儿子是2x，右儿子是2x+1，如1的左儿子是2，右儿子是3
堆有两个基本操作：down(x) up(x) (down( )和up( )最多只有一个会成立）
假设是一个小顶堆：

• 如何插入一个数(o(logn))：heap[++size] = x; up(size);
• 求集合当中的最小值(o(1))：heap[1]
• 删除一个值(o(logn))：用整个堆的最后一个元素（因为是用数组模拟的），覆盖掉堆顶的元素。操作：heap[1] = heap[size]; size–; down(1);
• 删除任意一个元素：heap[k] = heap[size]; size–; down(k)或up(k)
• 修改任意一个元素：heap[k] = x; down(k)或up(k)

down()是如何操作的？(o(logn))
从根节点开始，如果子节点比这个当前这个节点要小，则交换，把根节点换下去。
up()是如何操作的？(o(logn))

堆排序：把整个数组初始化为一个堆，每一次把堆顶输出（然后删掉）。
比如输出一个长度为n的整数数列从小到大的前m个小的数：

#include <iostream>
#include <algorithm>
using namespace std;

const int N = 100001;

int n, m;
int h[N], hsize;

void down(int u) {
	int min = u; 
    // 如果还存在左节点 且 左节点小于当前当前的根节点，则交换
    if (u * 2 <= hsize && h[u * 2] < h[min]) min = u * 2;
    // 如果右节点存在 且 右节点的值小于当前根节点
    if (u * 2 + 1 <= hsize && h[u * 2 + 1] < h[min]) min = u * 2 + 1;
    if (u != min) {
    	swap(h[u], h[min]);
        down(min);
    }
}

int main() {
	cin >> n >> m;
    hsize = n;
    for (int i = 1; i <= n; i++) cin >> h[i]; // 下标从1开始
    
    // o(n)的建堆方式
    for (int i = n / 2; i > 0; i --) down(i);

    while (m --) {
    	cout << h[1] << ' '; // 输出堆顶
        h[1] = h[hsize--];
        down(1);
    }
}