ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

模拟堆

2022-05-03 18:34:23  阅读:204  来源: 互联网

标签:cnt int hp swap ph 模拟 PM


维护一个集合,初始时集合为空,支持如下几种操作:

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

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
using namespace std;
const int N = 100010;
int h[N],cnt;//堆数组
int ph[N];//节点编号映射到堆数组下标
int 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 u){
    int t = u;
    if(2*u<=cnt&&h[t]>h[2*u]) t = 2*u;
    if(2*u+1<=cnt&&h[t]>h[2*u+1]) t = 2*u+1;
    if(u!=t){
        heap_swap(u,t);
        down(t);
    }
}
void up(int u)
{
    while (u / 2 && h[u] < h[u / 2])
    {
        heap_swap(u, u / 2);
        u /=2;
    }
}
int main(){
    int n;
    cin>>n;
    int idx=0;
    while(n--){
        string a;
        cin>>a;
        int k,x;
        if(a=="I"){
            cin>>x;
            cnt ++ ;//堆数组尾下标
            idx ++ ;//记录第几次插入
            ph[idx] = cnt, hp[cnt] = idx;
            h[cnt] = x;
            up(cnt);
        }
        else if(a=="PM")
            cout<<h[1]<<endl;
        else if(a=="DM"){
            heap_swap(1,cnt);
            cnt--;
            down(1);
        }
        else if(a=="D"){
            
            cin>>k;
            k=ph[k];//先将节点编号映射到堆数组下标
            heap_swap(k,cnt);
            cnt--;
            down(k);
            up(k);
        }
        else{
            cin>>k>>x;
            k=ph[k];
            h[k]=x;
            down(k);
            up(k);
        }
    }
    return 0;
}

https://www.acwing.com/solution/content/5661/

标签:cnt,int,hp,swap,ph,模拟,PM
来源: https://www.cnblogs.com/z218/p/16218945.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有