数据结构之堆

堆是一棵完全二叉树，分为大根堆和小根堆两种。

大根堆：根节点大于等于左右节点；小根堆：根节点小于等于左右节点；

大根堆的根节点是最大值，小根堆的根节点是最小值。

在c++中：

大根堆：

priority_queue<int,vector<int>,less<int>>q；

小根堆：

priority_queue<int,vector<int>,greater<int>>q。

操作：

q.size()，返回堆中的元素个数；q.top()，返回堆顶元素；q.push(x)，插入元素x；q.pop()，删除堆顶。

模拟堆：

对于完全二叉树：

假设根节点标号为 1 ，每个点的左儿子标号等于它的标号乘 2 ，右儿子标号等于它的标号乘2加1，故可以用一个一维数组来模拟堆。

改变堆中的一个数，要使堆依旧维持原有的性质，需要对其做上浮或下沉操作。

上浮操作：

将节点上的元素和其父节点上的元素比较，若该节点上的元素有小于其父节点上的的元素，则将两者中的元素调换，并递归处理该操作直到遍历到根节点。

下沉操作：

将节点上的元素和其两个子节点比较，若子节点中的元素有小于该节点上的元素，则将最小的子节点元素和该节点上的元素交换，并递归处理该操作直到遍历到叶节点。

图示：

依靠上浮和下沉这两个操作，可以对堆进行如下操作：

取出/删除堆顶：将堆中最后一个元素（数列中最后一个元素）覆盖到堆顶，然后下沉。

插入元素：向堆的最后面加入一个元素，然后上浮。

删除任意元素：将堆中最后一个元素覆盖到该位置，然后根据情况确定是上浮还是下沉。

堆的构造：

从最后一个非叶子节点开始，一直到根节点，每个节点执行一次下沉操作，就可以建成堆；

对于一个节点个数是n，根节点标号是1的完全二叉树，第一个非叶子节点的下标是n/2（下取整），即将n/2到1的节点一次进行down操作即可。

上浮操作和下沉操作的代码模板如下：

void down(int i)
{
    int t = i;
    if (i * 2 <= cnt && heap[i * 2] < heap[t])t = i * 2;
    if (i * 2 + 1 <= cnt && heap[i * 2 + 1] < heap[t])t = i * 2 + 1;
    if (heap[t] != heap[i]) {
        swap(heap[i], heap[t]);
        down(t);
    }
    return;
}


//不管是左儿子还是右儿子，除2向下取整都会得到根节点
void up(int i)
{
    int t = i;
    if (i / 2 >= 1 && heap[t] < heap[i / 2])t = i / 2;
    if (t != i) {
        swap(heap[t], heap[i]);
        up(t);
    }
    return;
}