洛谷 P1801 黑匣子_NOI导刊2010提高(06)对顶堆

题目描述

Black Box是一种原始的数据库。它可以储存一个整数数组,还有一个特别的变量i。最开始的时候Black Box是空的.而i等于0。这个Black Box要处理一串命令。

命令只有两种:

ADD(x):把x元素放进BlackBox;

GET:i加1,然后输出Blackhox中第i小的数。

记住:第i小的数,就是Black Box里的数的按从小到大的顺序排序后的第i个元素。例如:

我们来演示一下一个有11个命令的命令串。(如下图所示)

洛谷 P1801 黑匣子_NOI导刊2010提高(06)对顶堆

现在要求找出对于给定的命令串的最好的处理方法。ADD和GET命令分别最多200000个。现在用两个整数数组来表示命令串:

1.A(1),A(2),…A(M):一串将要被放进Black Box的元素。每个数都是绝对值不超过2000000000的整数,M$200000。例如上面的例子就是A=(3,1,一4,2,8,-1000,2)。

2.u(1),u(2),…u(N):表示第u(j)个元素被放进了Black Box里后就出现一个GET命令。例如上面的例子中u=(l,2,6,6)。输入数据不用判错。

输入输出格式

输入格式:

 

第一行,两个整数,M,N。

第二行,M个整数,表示A(l)

……A(M)。

第三行,N个整数,表示u(l)

…u(N)。

 

输出格式:

 

输出Black Box根据命令串所得出的输出串,一个数字一行。

 

输入输出样例

输入样例#1: 复制
7 4
3 1 -4 2 8 -1000 2
1 2 6 6
输出样例#1: 复制
3
3
1
2

说明

对于30%的数据,M≤10000;

对于50%的数据,M≤100000:

对于100%的数据,M≤200000。

 

此题类似于1168 中位数,思路都是对顶堆

对顶堆就是开一个大根堆和一个小根堆,大于大根堆堆首的就放入小根堆,以此维护某个序列的单调性

直接结合代码说吧

#include<bits/stdc++.h>
using namespace std;
const int MAXN=2e5+10;
priority_queue<int>p;//大根堆
priority_queue<int,vector<int>,greater<int> >q;//小根堆
int n,m,a[MAXN],b[MAXN],k,j;
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    for(int i=1;i<=m;i++)scanf("%d",&b[i]);
    for(int i=1;i<=m;i++)//对每个操作
    {
    while(j<b[i])我们不断放入元素,限制大根堆    
    {
        j++;
        p.push(a[j]);//先放入大根堆中
        q.push(p.top());//再把大根堆首放入小根堆中,这个操作保证
        p.pop();
    }
    printf("%d\n",q.top());
    p.push(q.top());
    q.pop();        
    }
    return 0;
}

 

上一篇:【tree】红黑树(下)


下一篇:设计模式-责任链模式