维护一个集合,初始时集合为空,支持如下几种操作:
I x
,插入一个数 x;PM
,输出当前集合中的最小值;DM
,删除当前集合中的最小值(数据保证此时的最小值唯一);D k
,删除第 k 个插入的数;C k x
,修改第 k 个插入的数,将其变为 x;
现在要进行 N 次操作,对于所有第 2 个操作,输出当前集合的最小值。
输入格式
第一行包含整数 N。
接下来 N 行,每行包含一个操作指令,操作指令为 I x
,PM
,DM
,D k
或 C k x
中的一种。
输出格式
对于每个输出指令 PM
,输出一个结果,表示当前集合中的最小值。
每个结果占一行。
数据范围
1≤N≤105
−109≤x≤109
数据保证合法。
输入样例:
8
I -10
PM
I -10
D 1
C 2 8
I 6
PM
DM
输出样例:
-10
6
在 heap_swap 中的三个 swap 不是和邻接表一样是相互存储,h,hp,ph之间两两存在映射关系 所以交换顺序的不同对结果并不会产生影响
m 用来记录插入数的个数
cnt 是记录堆中当前数据的多少
#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
int n, cnt, m;
int h[N], ph[N], hp[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 x){
int t = x;
if(2 * x <= cnt && h[2 * x] < h[t])
t = 2 * x;
if(2 * x + 1 <= cnt && h[2 * x + 1] < h[t])
t = 2 * x + 1;
if(t != x){
heap_swap(x, t);
down(t);
}
}
void up(int x){
while(x / 2 && h[x] < h[x / 2]){
heap_swap(x, x / 2);
x >>= 1;
}
}
int main(){
cin >> n;
while(n -- ){
string op;
int k, x;
cin >> op;
if(op == "I"){
cin >> x;
cnt ++ ;
m ++;
ph[m] = cnt;
hp[cnt] = m;
h[cnt] = x;
up(cnt);
}
else if(op == "PM"){
cout << h[1] << endl;
}
else if(op =="DM"){
heap_swap(1, cnt);
cnt -- ;
down(1);
}
else if(op == "D"){
cin >> k;
k = ph[k];
heap_swap(k, cnt);
cnt -- ;
up(k), down(k);
}
else{
cin >> k >> x;
k = ph[k];
h[k] = x;
up(k), down(k);
}
}
return 0;
}