题目:模拟堆
维护一个集合,初始时集合为空,支持如下几种操作:
“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≤10^5
−109≤x≤109
数据保证合法。
输入样例
8
I -10
PM
I -10
D 1
C 2 8
I 6
PM
DM
输出样例
-10
6
思路:
首先需要知道第k个修改的数字在哪里 然后才能对其进行修改
ph[K]存的是 第k个插入的点是哪一个点
hp[k]存的是 堆里面的点是第几个插入的点
ph[j] = k 第j个插入的点在堆里面的下标是k
hp[k] = j 堆里面下表是k的点对应的ph数组的下标是j
//因为我既要从第几个插入的点找到是堆里面的第几个元素
又要 通过堆里面的元素找出是第几个插入的点
代码:
#include <iostream>
#include <string.h>
using namespace std;
const int N = 100010;
//ph表示从下标映射到堆里面 p-> 下标 h-> 堆
int hp[N], ph[N];
int se;//堆的大小
int h[N];
void head_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)
{
int t = u;
if (u * 2 <= se && h[u * 2] < h[u])
t = u * 2;
if (u * 2 + 1 <= se && h[u * 2 + 1] < h[u])
t = u * 2 + 1;
if (u != t)
{
head_swap(u,t);
down(t);
}
}
void up(int u)
{
if (u / 2 != 0 && h[u / 2] >= h[u])
{
head_swap(u/2,u);
u /= 2;
}
}
int main()
{
int x,m = 0;//m表示记录第几个插入的数字
int n; cin >> n;
while (n--)
{
char op[10];
scanf("%s",op);
if (!strcmp(op,"I"))
{
cin >> x;
se++;
h[se] = x;
ph[m] = se, hp[se] = m;
h[se] = x;
up(se);
}
else if (!strcmp(op,"PM"))
{
cout << h[1];
}
else if (!strcmp(op,"DM"))
{
head_swap(1,se);
se--;
down(1);
}
else if (!strcmp(op, "D"))
{
int k; cin >> k;
k = hp[k];
head_swap(k,se);
se--;
down(k);
up(k);
}
else {
int k, x;
cin >> k >> x;
k = ph[k];
h[k] = x;
down(k), up(k);
}
}
return 0;
}