839. 模拟堆
维护一个集合,初始时集合为空,支持如下几种操作:
I x
,插入一个数 xx;PM
,输出当前集合中的最小值;DM
,删除当前集合中的最小值(数据保证此时的最小值唯一);D k
,删除第 kk 个插入的数;C k x
,修改第 kk 个插入的数,将其变为 xx;
现在要进行 NN 次操作,对于所有第 22 个操作,输出当前集合的最小值。
输入格式
第一行包含整数 NN。
接下来 NN 行,每行包含一个操作指令,操作指令为 I x
,PM
,DM
,D k
或 C k x
中的一种。
输出格式
对于每个输出指令 PM
,输出一个结果,表示当前集合中的最小值。
每个结果占一行。
数据范围
1≤N≤1051≤N≤105
−109≤x≤109−109≤x≤109
数据保证合法。
输入样例:
8
I -10
PM
I -10
D 1
C 2 8
I 6
PM
DM
输出样例:
-10
6
#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+10;
int h[N],ph[N],hp[N],cnt;
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 up(int u)
{
while(u/2&&h[u]<h[u/2]){
heap_swap(u,u/2);
u=u/2;
}
}
void down(int l)
{
int t=l;
if(l*2<=cnt&&h[l*2]<h[t]) t=l*2;
if(l*2+1<=cnt&&h[l*2+1]<h[t]) t=l*2+1;
if(l!=t){
heap_swap(l,t);
down(t);
}
}
int main()
{
int n,m=0;
scanf("%d", &n);
while (n -- )
{
string s;
cin>>s;
int x,y;
if(s=="I")
{
scanf("%d", &y);
cnt++,m++;
ph[m]=cnt;hp[cnt]=m;
h[cnt]=y;
up(cnt);
}
else if(s=="PM") printf("%d\n",h[1]);
else if(s=="DM")
{
heap_swap(1,cnt);
cnt--;
down(1);
}
else if(s=="D")
{
scanf("%d", &x);
x=ph[x];
heap_swap(x,cnt);
cnt--;
up(x);down(x);
}
else
{
scanf("%d%d", &x, &y);
x=ph[x];h[x]=y;
up(x);down(x);
}
}
return 0;
}