算法板子

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int h[N], ph[N], hp[N];
int Size, m, n;
void heap_swap(int a, int b) {
    swap(ph[hp[a]], ph[hp[b]]);
    swap(hp[a], hp[b]);
    swap(h[a], h[b]);
}

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

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

int main() {
    cin >> m;
    while(m--) {
        char op[10];
        scanf("%s", &op);
        if (!strcmp(op, "I")) {
            int x;
            scanf("%d", &x);
            ++Size;
            ++n;
            h[Size] = x;
            ph[n] = Size;
            hp[Size] = n;
            up(Size);
        } else if (!strcmp(op, "PM")) {
            printf("%d\n", h[1]);
        } else if (!strcmp(op, "DM")) {
            heap_swap(1, Size);
            --Size;
            down(1);
        } else if (!strcmp(op, "D")) {
            int k;
            scanf("%d", &k);
            k = ph[k];
            heap_swap(Size, k);
            --Size;
            down(k), up(k);
        } else if (!strcmp(op, "C")) {
            int k, x;
            scanf("%d%d", &k, &x);
            k = ph[k];
            h[k] = x;
            down(k), up(k);
        }
    }
    
    return 0;
}
本文为原创文章,欢迎分享,勿全文转载,如果内容你实在喜欢,可以加入收藏夹,说不定哪天故事又继续更新了呢。
本文地址:https://qoogle.top/heap-implementation/
最后修改日期:2020年7月3日

作者

留言

撰写回覆或留言

发布留言必须填写的电子邮件地址不会公开。