C++算法题 # 31 模拟堆

题目描述
维护一个集合，初始时集合为空，支持如下几种操作：

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≤10^9
数据保证合法。

输入样例：

8
I -10
PM
I -10
D 1
C 2 8
I 6
PM
DM

输出样例：

-10
6

思路：

这里实现了上面的5个功能。由于本题的k时第几次插入的节点，因此本题需要两个额外的数组来存放插入次序和节点位置的映射关系。

但这里要注意一点就是正确区分hp[N] 和 ph[N] 两个数组的区别，有点绕，容易搞错。

代码示例

#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1e5 + 10;

int n;
int h[N], siz, t;
int hp[N], ph[N];// ph[N] 存放第k次插入的点当前位置 hp[k] 第k个位置插入的点为第几次插入的

// 修改交换函数，应为交换两个点的话，对应的hp 和 ph 数组存放的值也要修改
void h_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 min_id = x;
	if(x * 2 <= siz && h[x * 2] < h[min_id]) min_id = x * 2;
	if(x * 2 + 1 <= siz && h[x * 2 + 1] < h[min_id]) min_id = x * 2 + 1;
	if( min_id != x)
	{
		h_swap(x,min_id);
		down(min_id);
	}
	
	
}
// 将当前节点的值向上调整，直至堆稳定
void up(int x)
{
	// x / 2 为当前节点的父节点
	while( x / 2 && h[x / 2] > h[x])
	{
		h_swap(x / 2, x);
		x /= 2;
	}

}


int main()
{
	cin >> n;
	
	
	string op;
	int k, k1, x;
	while(n --)
	{
		cin >> op;
		
		if(op == "I")
		{
			cin >> x;
			h[++ siz] = x;
			hp[siz] = ++ t;
			ph[t] = siz;
			up(siz);
	
		}
		else if(op == "PM")
		{
			cout << h[1] << endl;
		}
		else if(op == "DM")
		{
			h_swap(1, siz);// 这里也要交换所有的相关数组 包括h hp 和 ph
			siz --;
			down(1);
			
			
		}
		else if(op == "D")
		{
			cin >> k1;
			k = ph[k1];
			h_swap(k, siz);// 这里就要进行所有的交换了
			siz --;
			up(k);
			down(k);
			
		}
		else
		{
			cin >> k1 >> x;
			k = ph[k1];
			h[k] = x;
			up(k);
			down(k);
			
	
		}
		
	}
	
	
	
	
	
	return 0;
}