Luogu3378 【手写堆模板】

题目描述

给定一个数列，初始为空，请支持下面三种操作：

1. 给定一个整数 xx，请将 xx 加入到数列中。
2. 输出数列中最小的数。
3. 删除数列中最小的数（如果有多个数最小，只删除 11 个）。

输入格式

第一行是一个整数，表示操作的次数 nn。
接下来 nn 行，每行表示一次操作。每行首先有一个整数 opop 表示操作类型。

• 若 op = 1op=1，则后面有一个整数 xx，表示要将 xx 加入数列。
• 若 op = 2op=2，则表示要求输出数列中的最小数。
• 若 op = 3op=3，则表示删除数列中的最小数。如果有多个数最小，只删除 11 个。

输出格式

对于每个操作 22，输出一行一个整数表示答案。

输入输出样例

输入 #1复制

5
1 2
1 5
2
3
2

输出 #1复制

2
5

说明/提示

数据规模与约定

• 对于 30\%30% 的数据，保证 n \leq 15n≤15。
• 对于 70\%70% 的数据，保证 n \leq 10^4n≤104。
• 对于 100\%100% 的数据，保证 1 \leq n \leq 10^61≤n≤106，1 \leq x \lt 2^{31}1≤x<231，op \in \{1, 2, 3\}op∈{1,2,3}。

 

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <string>
using namespace std;

const int N = 1000001;
int heap[N], n, ans;
int tot;

void down(int k){
    int l = k*2;
    int r = k*2 + 1;
    if(l > tot) return;
    if(heap[l] < heap[k] || heap[r] < heap[k]){
        if(heap[l] < heap[r]){
            swap(heap[l], heap[k]);
            down(l);
        }else{
            swap(heap[r], heap[k]);
            down(r);
        }
    }
}

void up(int k){
    if(k == 1)
        return;
    if(heap[k] < heap[k>>1]){
        swap(heap[k], heap[k>>1]);
        up(k>>1);
    }
}

void push(int x){
    heap[++tot] = x;
    up(tot);
}

int pop(){
    int x = heap[1];
    heap[1] = heap[tot--];
    down(1);
    return x;
}

int main(){
    memset(heap, 127, sizeof(heap));
    scanf("%d", &n);
    for(int i=1;i<=n;i++){
        int x;
        scanf("%d", &x);
        if(x == 1){
            int tmp;
            scanf("%d", &tmp);
            push(tmp);
        }
        if(x == 2){
            printf("%d\n", heap[1]);
        }
        if(x == 3){
            pop();
        }
            
    }
    
    return 0;
}