[堆+模板] 模拟堆模板

文章目录

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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Ypuyu

如果帮助到你,可以请作者喝水~

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值