本蒟蒻开始学堆了
题目描述
如题,初始小根堆为空,我们需要支持以下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;
}
// 大根堆 / 小根堆 模板
还有用优先队列来实现与堆类似的效果的方法,但是貌似慢很多。