【模板】堆

本蒟蒻开始学堆了

题目描述

如题，初始小根堆为空，我们需要支持以下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

来源：Luogu P3378

首先来一份手写的丑的一比的堆。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int inf=0x7f7f7f7f;
int d[1000010],n,x,y,l=1;
int read()
{
    int p=0,ff=1;char c=getchar();
    while(c<'0'||c>'9') {if(c=='-')ff=-1;c=getchar();}
    while(c>='0'&&c<='9') {p=p*10+c-'0';c=getchar();}
    return p*ff;
}
int mm(int a,int b)
{
    if(d[a]<d[b]) return a;
    else return b; 
}
inline void add()
{
    while(d[l]<inf) l++;
    d[l]=y;
    int i=l;
    while(d[(i>>1)]>d[i]&&i!=1)
    {
        swap(d[(i>>1)],d[i]);
        i=(i>>1);
    }
}
inline void del()
{
    d[1]=inf;
    int i=1;
    int k=mm((i<<1),(i<<1)+1);
    while(d[i]>d[k])
    {
        swap(d[i],d[k]);
        i=k;
        k=mm((i<<1),(i<<1)+1);
    }
}
int main()
{
    memset(d,inf,sizeof(d));
    n=read();
    for(int j=1;j<=n;j++)
      {
          x=read();
          if(x==1) 
            {
                y=read();
                add();
          }
        if(x==3) del();
        if(x==2) printf("%d\n",d[1]);
      }
    return 0;
}

再来一份简洁的STL

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int d[1000010],n,x,y,num;
int read()
{
    int p=0,ff=1;char c=getchar();
    while(c<'0'||c>'9') {if(c=='-')ff=-1;c=getchar();}
    while(c>='0'&&c<='9') {p=p*10+c-'0';c=getchar();}
    return p*ff;
}
int main()
{
    n=read();
    for(int j=1;j<=n;j++)
      {
          x=read();
          if(x==1) 
            {
                d[++num]=read();
                push_heap(d+1,d+num+1,greater/*小根堆. less 为大根堆*/<int>());
          }
        if(x==3) 
          {
              pop_heap(d+1,d+num+1,greater<int>()); 
              num--;
          }
        if(x==2) printf("%d\n",d[1]);
      }
    return 0;
}
// 大根堆 / 小根堆 模板

还有用优先队列来实现与堆类似的效果的方法，但是貌似慢很多。