stl 优先队列(堆)

Posted 2020-10-02 1 2 f s_____

【模板】堆

题目描述

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

#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
priority_queue<int,vector<int>,greater<int> >q;
int main()
{
    int n;
    scanf("%d",&n);
    int pd;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&pd);
        if(pd==1) 
        {
            int v;
            scanf("%d",&v);
            q.push(v);
        }
        if(pd==2)
        {
            printf("%d\n",q.top());
        }
        if(pd==3)
        {
            q.pop();
        }
    }
    return 0;
}

View Code

一直不会stl堆（以前一直手写堆），今天终于学会了，谢谢cyc的指导；

下来是一道运用stl堆的题----合并果子；

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
int n;
priority_queue<int,vector<int>,greater<int> >q;
int map[10010];
int main()
{
    scanf("%d",&n);
    int ans=0;
    for(int i=1;i<=n;i++) scanf("%d",&map[i]),q.push(map[i]);
    for(int i=2;i<=n;i++)
    {
        int v=q.top();q.pop();
        int u=q.top();q.pop();
        ans+=v+u;
        q.push(v+u);
    }
    printf("%d",ans);
    return 0;
}

View Code