洛谷 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)个元素被放进了Blaek 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
l
2

说明

对于30%的数据，M≤10000;
对于50%的数据，M≤100000：
对于100%的数据，M≤200000。

分析

先s,代表题目中的i
再设两个堆,一个大根堆,一个小根堆,
大根堆里放目前s个最小的数,
小根堆里放剩下的数
具体过程:
那么可知大根堆里必定有s个数,
那么小根堆里的第一个数就是s+1也就是下一个get所需要的值
那么就不断地往大根堆里添加值,当超过s时就移到小根堆里去,
碰到get是就不断输出,再不断把输出的值在返还到大根堆里重新在按上面来一遍即可,
然后c++自带堆。。。

代码

#include <bits/stdc++.h>

using namespace std;

int a[200001],u[200001];

priority_queue<int,vector<int>,greater<int>> c;
priority_queue<int> b;

int main()
{
    int m,n;
    int s,k=1;
    cin>>m>>n;
    for (int i = 1; i <= m; i++)
    {
        cin>>a[i];
    }
    for (int i = 1; i <= n; i++)
    {
        cin>>u[i];
    }
    for (int i = 1; i <= m; i++)
    {
        b.push(a[i]);
        if (b.size() > s)
        {
            c.push(b.top());
            b.pop(); 
        }
        while (u[k] == i)
        {
            s++;
            cout<<c.top()<<"\n";
            b.push(c.top());
            c.pop();
            k++;
        }
    }
}