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个命令的命令串。（如下图所示）

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

题解:对于堆来说，只能求出当前序列最小的，那么对于这道题我们可以维护两个堆，一个大顶堆和一个小顶堆，大顶堆里面维护i个最小的元素，小顶堆则维护剩下的元素，那么此时小顶堆的堆顶元素就是第i+1小的元素，对于每次添加元素的时候，先考虑是否大于大顶堆的堆顶元素，若大于则将其加入小顶堆，否则就将大顶堆的堆顶元素弹出到小顶堆，再加入大顶堆中。输出则直接输出小顶堆的堆顶元素，然后将堆顶元素加入大顶堆即可。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e6+7;

void read(int &x){scanf("%d",&x);}
int a[maxn];

int main()
{
    int n;
    scanf("%d",&n);

    priority_queue<int,vector<int>,less<int> >q1;
    priority_queue<int,vector<int>,greater<int> >q2;
    int t;
    read(t);

    int len=0;
    for(int i=1;i<=n;i++)
        read(a[i]);

    int st=1;
    for(int i=1;i<=t;i++)
    {
        int ti;
        read(ti);

        for(;st<=ti;st++)
        {
            if(i==1)
                q2.push(a[st]);
            else
            {
                if(a[st]<q1.top())
                {
                    q2.push(q1.top());
                    q1.pop();
                    q1.push(a[st]);
                }
                else
                q2.push(a[st]);
            }
        }
        printf("%d\n",q2.top());
        q1.push(q2.top());
        q2.pop();
    }

    return 0;
}