AcWing 839. 模拟堆

Posted 2022-02-11 MangataTS

题目链接

https://www.acwing.com/problem/content/841/

思路

纯纯模拟堆的操作，后面细说，可以先看代码

代码

#include<bits/stdc++.h>
using namespace std;
//----------------自定义部分----------------
#define ll long long
#define mod 1000000009
#define endl "\\n"
#define PII pair<int,int>

int dx[4]=0,-1,0,1,dy[4]=-1,0,1,0;

ll ksm(ll a,ll b) 
	ll ans = 1;
	for(;b;b>>=1LL) 
		if(b & 1) ans = ans * a % mod;
		a = a * a % mod;
	
	return ans;


ll lowbit(ll x)return -x & x;

const int N = 2e6+10;
//----------------自定义部分----------------
int n,m,cnt,h[N],hp[N],ph[N];
//ph记录的是第i次存入的下标
//hp记录的是下标为i是第几次存入的
//ph和hp是一个互逆的数组

void heap_swap(int a,int b)//a、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 <= cnt && h[u * 2] < h[t]) t = u * 2;
	if((u * 2 + 1) <= cnt && h[u * 2 + 1] < h[t]) t = u * 2 + 1;
	if(u != t) 
		heap_swap(u,t);
		down(t);
	


void up(int u) //向上更新
	while((u>>1) && h[u] < h[u>>1]) 
		heap_swap(u,u >> 1);
		u >>= 1;
	


int main()

	std::ios::sync_with_stdio(false);
	std::cin.tie(0);
	std::cout.tie(0);
	string op;
	cin>>n;
	int k,x,kk=0;
	for(int i = 1;i <= n; ++i)
		cin>>op;
		if(op == "I")//在堆中插入元素
			cin>>k;
			cnt++;
			kk++;//表示是第kk次插入的
			h[cnt] = k;
			ph[kk] = cnt,hp[cnt] = kk;
			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")//删除第k次插入的数
			cin>>k;
			k = ph[k];//先找到第k个插入的下标
			heap_swap(k,cnt);//然后就是和删除头节点类似的操作，即交换最后一个元素，然后做一次down
			cnt--;
			down(k),up(k);
		
		else
			cin>>k>>x;//更改第k次插入的数
			k = ph[k];//还是先找到第k个插入的下标
			h[k] = x;//然后将改下标的值更改为x
			down(k),up(k);//这个更改的值要么大于等于原来的h[k]，要么小于等于，所以我们直接down和up就好了
		
	 
	
	return 0;