[堆+模板] 模拟堆模板

0. 前言

Biu

堆优化在某些问题中是关键的一步，比如应用在 堆优化版 Dijkstra 算法，不过大多用的是 STL 中的 priority_queue 了，数组模拟堆也是挺重要的。

这个板子在普通小顶堆的基础上，支持删除和修改第 k 个插入的数，故维护的信息多了两个数组，用以维护第 j 个插入点在堆中的下标是多少，以及堆中第 k 个点是第几个被插入的。其实这两个数组维护信息是互逆的。

代码上该注意的点都有过注释了。至于堆也就 down()、up() 操作，至于修改以及删除是采用了元素覆盖的方法进行替换删除的，最后就是不知道修改、删除后应该执行 down() 还是 up() 时，就两个都执行一遍，反正代码只会挑其中一个执行，相当于简化了代码。

1. 模拟堆

模板代码：

#include <iostream>
#include <string.h>

using namespace std;

const int N = 1e5+5;

int n, m;
// h存堆中元素数值
// ph[j]=k,表示第j个插入点在堆中的下标是k
// hp[k]=j,表示堆中第k个点对应的ph数组的下标是j，两者互逆 
// 因为需要从插入点找到堆中元素，又需要从堆中元素找回插入点
int h[N], ph[N], hp[N], tt; 

void heap_swap(int a, int b) {
    swap(ph[hp[a]], ph[hp[b]]); // 堆中的下标交换，hp[a]是堆中的下标对应到ph的下标
    swap(hp[a], hp[b]);         // hp由于堆中下标变换了，所以也需要跟着改变一下，指向对应的ph数组
    swap(h[a], h[b]);           // 值交换
}

void down(int u) {
    int t = u;
    if (u * 2 <= tt && h[u * 2] < h[t]) t = u * 2;
    if (u * 2 + 1 <= tt && h[u * 2 + 1] < h[t]) t = u * 2 + 1;
    if (u != t) heap_swap(u, t), down(t);
}

void up(int u) {
    while (u / 2 && h[u / 2] > h[u]) heap_swap(u / 2, u), u /= 2;
}

int main() {
    cin >> n;
    while (n --) {
        char op[10];
        int k, x;
        cin >> op;
        if (!strcmp(op, "I")) {         // 插入操作
            cin >> x; 
            tt++, m++;                  // m为插入的次数
            ph[m] = tt;                 // 第m个插入的数，在堆中就是tt位置
            hp[tt] = m;                 // 堆中tt位置的数就是第m个插入的数
            h[tt] = x;
            up(tt);
        } 
        else if (!strcmp(op, "PM")) cout << h[1] << endl;       // 输出最小值
        else if (!strcmp(op, "DM")) {                           // 删除最小值
            heap_swap(1, tt);
            tt--;
            down(1);
        } 
        else if (!strcmp(op, "D")) {                            // 删除第k个插入的数
            cin >> k;
            k = ph[k];                                          // ph[k]是第k个数在堆中的位置
            heap_swap(k, tt);
            tt--;
            down(k), up(k);
        }
        else {                                                  // 修改第k个插入的数
            cin >> k >> x;
            k = ph[k];
            h[k] = x;
            down(k), up(k);
        }
    }
    return 0;
}