【模板】二叉堆 （模板题：洛谷P3378堆）

题目描述

如题，初始小根堆为空，我们需要支持以下3种操作：

操作1： 1 x 表示将x插入到堆中

操作2： 2 输出该小根堆内的最小数

操作3： 3 删除该小根堆内的最小数

输入输出格式

输入格式：

第一行包含一个整数N，表示操作的个数

接下来N行，每行包含1个或2个正整数，表示三种操作，格式如下：

操作1： 1 x

操作2： 2

操作3： 3

输出格式：

包含若干行正整数，每行依次对应一个操作2的结果。

输入输出样例

输入样例#1：

5
1 2
1 5
2
3
2

输出样例#1：

2
5

说明

时空限制：1000ms,128M

数据规模：

对于30%的数据：N<=15

对于70%的数据：N<=10000

对于100%的数据：N<=1000000（注意是6个0。。。不过不要害怕，经过编者实测，堆是可以AC的）

样例说明：

故输出为2、5

#include <iostream>
#include <cstdio>
using namespace std;

int n=0,m,flag,x;
int a[3000000];

void heapify(int t){     //维护小顶堆 
    int l=t<<1;
    int r=t<<1|1;
    int mini=t;
    if (l<=n)  mini= a[l]<a[mini]? l:mini;
    if (r<=n)  mini= a[r]<a[mini]? r:mini;
    if (mini!=t){
        int tt=a[t];a[t]=a[mini];a[mini]=tt;
        heapify(mini);
    }
}

void insert(int t){     //插入元素 
    ++n;a[n]=t;
    int p=n;
    while (p>1 && a[p/2]>t){  //迭代 
        a[p]=a[p/2];          //找到合适的位置 
        p=p/2;                //此处将比它大的 
    }                         //元素下沉 
    a[p]=t;
}

void getminimum(){             //删除堆顶元素 
    int maxi=a[1];
    a[1]=a[n];
    --n;
    heapify(1);
}


int main(){
    scanf("%d",&m);
    for (int i=1;i<=m;++i){
        scanf("%d",&flag);
        if (flag==1){scanf("%d",&x);insert(x);}
        if (flag==2){printf("%d\n",a[1]);}
        if (flag==3){getminimum();}
    }
    return 0;
}