黑匣子（优先队列，大小根堆）

黑匣子

（NOI导刊2010提高（06））

（来自Luogu）

题目描述
Black Box是一种原始的数据库。它可以储存一个整数数组，还有一个特别的变量i。最开始的时候Black Box是空的．而i等于0。这个Black Box要处理一串命令。
命令只有两种：
ADD(x):把x元素放进BlackBox;
GET:i加1，然后输出Blackhox中第i小的数。
记住：第i小的数，就是Black Box里的数的按从小到大的顺序排序后的第i个元素。例如：
我们来演示一下一个有11个命令的命令串。（如下图所示）

现在要求找出对于给定的命令串的最好的处理方法。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。

思路

这个题是利用STL库里面的优先队列来做的，我们可以考虑线性结构上有两个堆，左边是大根堆，右边是小根堆，这样我们确定一个数是通过让大根堆和小根堆的堆顶所在的位置同时向左或向右移动。
而向右移动的时候我们让大根堆堆顶压入小根堆，然后pop掉大根堆中的这个元素，向左移动的话就是让小根堆堆顶压入大根堆，然后pop掉小根堆中的这个元素。
然后读出一个固定的根堆的堆顶元素即可。
这个题目类似于一个重庆一次省队选拔赛的一道叫做中位数的题目，可以试着用这个方法AC那个题目。
还有就是要熟练背诵这段STL的代码。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<queue>
using namespace std;
long long i,j,m,n;
long long a[200001],b[200001];

priority_queue<long long> l;//da
priority_queue<long long, vector<long long>, greater<long long> > r;//xiao
long long rr()
{
    long long ans=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')
        f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        ans*=10;
        ans+=ch-'0';
        ch=getchar();
    }
    return ans*f;
}

int main()
{
    n=rr(),m=rr();
    for(i=1;i<=n;i++)
    a[i]=rr();

    for(i=1;i<=m;i++)
    b[i]=rr();

    long long pp=1,mid=0;
    for(i=1;i<=n;i++)
    {
        if(pp>m) break;
        if(l.empty()||l.top()>=a[i])
        l.push(a[i]),mid++;
        else
        r.push(a[i]);
        while(b[pp]==i)
        {
            while(pp>mid)             
            {
                l.push(r.top());
                r.pop();
                mid++;
            }
            while(pp<mid)
            {
                r.push(l.top());
                l.pop();
                mid--;
            }
            cout<<l.top()<<endl;
            pp++;
        }
    }

    return 0;
}