【洛谷】P3378 【模板】堆

题目地址：

https://www.luogu.com.cn/problem/P3378

题目描述：
给定一个数列，初始为空，请支持下面三种操作：
1.给定一个整数$x$，请将$x$加入到数列中。
2.输出数列中最小的数。
3.删除数列中最小的数（如果有多个数最小，只删除$1$个）。

输入格式：
第一行是一个整数，表示操作的次数$n$。
接下来$n$行，每行表示一次操作。每行首先有一个整数$op$表示操作类型。
若$op=1$，则后面有一个整数$x$，表示要将$x$加入数列。
若$op=2$，则表示要求输出数列中的最小数。
若$op=3$，则表示删除数列中的最小数。如果有多个数最小，只删除$1$个。

输出格式：
对于每个操作$2$，输出一行一个整数表示答案。

数据范围：
对于$30\%$的数据，保证$n\le 15$。
对于$70\%$的数据，保证$n\le 10^4$。
对于$100\%$的数据，保证$1\le n\le 10^6$，$1\le x<2^{31}$， o p ∈ { 1 , 2 , 3 } op \in \{1, 2, 3\} op∈{1,2,3}。

可以用二叉堆实现，代码如下：

#include <iostream>
using namespace std;

const int N = 1e6 + 10;
int n;
int h[N], sz;

void sift_up(int k) {
  while (k > 1 && h[k] < h[k / 2]) {
    swap(h[k], h[k / 2]);
    k /= 2;
  }
}

void sift_down(int k) {
  while (k * 2 <= sz) {
    int t = k;
    if (h[k * 2] < h[t]) t = k * 2;
    if (k * 2 + 1 <= sz && h[k * 2 + 1] < h[t]) t = k * 2 + 1;
    if (t == k) return;
    swap(h[k], h[t]);
    k = t;
  }
}

void push(int x) {
  h[++sz] = x;
  sift_up(sz);
}

void pop() {
  h[1] = h[sz--];
  sift_down(1);
}

int main() {
  scanf("%d", &n);
  while (n--) {
    int op, x;
    scanf("%d", &op);
    if (op == 1) {
      scanf("%d", &x);
      push(x);
    } else if (op == 2) printf("%d\n", h[1]);
    else pop();
  }
}

每次操作时间复杂度$O(\log n)$，$n$为数组中有多少个数，空间$O(n)$。