模拟堆空间超了,怎么调,求助
  • 板块学术版
  • 楼主XSean
  • 当前回复1
  • 已保存回复1
  • 发布时间2022/11/13 11:33
  • 上次更新2023/10/27 03:08:12
查看原帖
模拟堆空间超了,怎么调,求助
546830
XSean楼主2022/11/13 11:33

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

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 数据保证合法。

#include <bits/stdc++.h>
//chao
//dui mo ni
using namespace std;
const int N = 100010;
int h[N], hp[N]/*堆中下标为k的点是第几个插入的*/, ph[N]/*第k个插入的点在堆中是第几个*/, 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 down(int u){
    int t = u;
    if(u * 2 <= cnt && h[t] > h[u * 2]) t = u * 2;
    if(u * 2 + 1 <= cnt && h[t] > h[u * 2 + 1]) t = u * 2 + 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 >>= 1;
    }
}
int main(){
    int n, m = 0/*第几个插入*/;
    scanf("%d", &n);
    
    while(n--){
        char ins[3];
        int k, x;
        scanf("%s", &ins);
        if(!strcmp(ins, "I")){
            scanf("%d", &x);
            cnt++;
            m++;
            ph[m] = cnt, hp[cnt] = m;
            h[cnt] = x;
            up(cnt);
        }else if(!strcmp(ins, "PM")) printf("%d\n", h[1]);
        else if(!strcmp(ins, "DM")){
            heap_swap(1, cnt);
            cnt--;
            down(1);
        }else if(!strcmp(ins, "D")){
            scanf("%d", &k);
            k = ph[k];
            heap_swap(cnt, k);
            cnt--;
            down(k), up(k);
        }else{
            scanf("%d%d", &k, &x);
            k = ph[k];
            h[k] = x;
            down(k), up(x);
        }
    }
    return 0;
}
2022/11/13 11:33
加载中...