AcWing 839. 模拟堆（二叉堆）

思路

和 set 的用途差不多。

代码

#include <bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int h[N], ph[N], hp[N], sz, m;          //m 表示被插入的数的计数器

void heap_swap(int a, int b)
{
    
    
    swap(h[a], h[b]);
    swap(ph[hp[a]], ph[hp[b]]);
    swap(hp[a], hp[b]);
}

void down(int u)
{
    
    
    int t = u;
    if (u * 2 <= sz && h[t] > h[u * 2]) t = u * 2;
    if (u * 2 + 1 <= sz && h[t] > h[u * 2 + 1]) t = u * 2 + 1;
    if (u != t)
    {
    
    
        heap_swap(u, t);
        down(t);
    }
}
void up(int u)
{
    
    
    while (u / 2 > 0 && h[u] < h[u / 2])
    {
    
    
        heap_swap(u, u / 2);
        u = u / 2;
    }
}

int main()
{
    
    
    int n; cin >> n;
    string op; int k, x;
    while (n --)
    {
    
    
        cin >> op;
        if (op == "I") 
        {
    
    
            cin >> x;
            h[++ sz] = x;
            ph[++ m] = sz;
            hp[sz] = m;
            up(sz);
        }
        else if (op == "PM")
        {
    
    
            cout << h[1] << endl;

        }
        else if (op == "DM")
        {
    
    
            heap_swap(1, sz);
            sz --;
            down(1);
        }
        else if (op == "D")
        {
    
    
            cin >> k;
            k = ph[k];              //注意这里是删除第 k 个插入的那个数，所以我们通过ph[] 来获取的这个数在数组中的下表为 ph[k]
            heap_swap(k, sz);
            sz --;
            down(k);
            up(k);
        }
        else if (op == "C")
        {
    
    
            cin >> k >> x;
            k = ph[k];
            h[k] = x;
            down(k);
            up(k);
        }
    }

    return 0;
}