思路
和 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;
}