P1801 黑匣子

黑匣子

题目描述

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

命令只有两种：

• ADD(x)：把 $x$ 元素放进 Black Box;

• GET：$i$ 加 $1$，然后输出 Black Box 中第 $i$ 小的数。

记住：第 $i$ 小的数，就是 Black Box 里的数的按从小到大的顺序排序后的第 $i$ 个元素。

我们来演示一下一个有11个命令的命令串。（如下表所示）

序号	操作	$i$	数据库	输出
1	ADD(3)	$0$	$3$	/
2	GET	$1$	$3$	$3$
3	ADD(1)	$1$	$1,3$	/
4	GET	$2$	$1,3$	$3$
5	ADD(-4)	$2$	$-4,1,3$	/
6	ADD(2)	$2$	$-4,1,2,3$	/
7	ADD(8)	$2$	$-4,1,2,3,8$	/
8	ADD(-1000)	$2$	$-1000,-4,1,2,3,8$	/
9	GET	$3$	$-1000,-4,1,2,3,8$	$1$
10	GET	$4$	$-1000,-4,1,2,3,8$	$2$
11	ADD(2)	$4$	$-1000,-4,1,2,2,3,8$	/

现在要求找出对于给定的命令串的最好的处理方法。ADD 命令共有 $m$ 个，GET 命令共有 $n$ 个。现在用两个整数数组来表示命令串：

1. $a_1,a_2,\cdots ,a_m$：一串将要被放进 Black Box 的元素。例如上面的例子中 $a=[3,1,-4,2,8,-1000,2]$。

2. $u_1,u_2,\cdots ,u_n$：表示第 $u_i$ 个元素被放进了 Black Box 里后就出现一个 GET 命令。例如上面的例子中 $u=[1,2,6,6]$ 。输入数据不用判错。

输入格式

第一行两个整数 $m$ 和 $n$，表示元素的个数和 GET 命令的个数。

第二行共 $m$ 个整数，从左至右第 $i$ 个整数为 $a_i$，用空格隔开。

第三行共 $n$ 个整数，从左至右第 $i$ 个整数为 $u_i$，用空格隔开。

输出格式

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

样例 #1

样例输入 #1

7 4
3 1 -4 2 8 -1000 2
1 2 6 6

样例输出 #1

3
3
1
2

提示

数据规模与约定

• 对于 $30\%$ 的数据，$1\le n,m\le 10^4$。
• 对于 $50\%$ 的数据，$1\le n,m\le 10^5$。
• 对于 $100\%$ 的数据，$1\le n,m\le 2\times 10^5,|a_i|\le 2\times 10^9$，保证 $u$ 序列单调不降。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define de(x) cout<<x<<" ";
#define sf(x) scanf("%d",&x);
#define Pu puts("");
const int N=2e5+10;
priority_queue<int>qx;//大根堆 即最先pop的是最大的元素
priority_queue<int,vector<int>,greater<int> >qn;//小根堆
int n,m;
int a[N];
int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        sf(a[i])
    }
    //第一行输入a[i]
    //第二行输入q[j]=x
    //本题是求解a数组的前x个元素中第j大的元素
    int q,r=1;
    for(int i=1;i<=m;i++){
        sf(q)
        for(int j=r;j<=q;j++){
            qx.push(a[j]);
            if(qx.size()==i){
    //要求解第i大的元素ans

    //此处是精髓：对于每个元素，刚开始都直接放入大顶堆x中
    //当元素个数达到i时，就把元素放到小顶堆y中,保证前最小的i-1个元素在x中
    //当for循环结束时，y中的元素包括了所求解的第i大元素ans,且它是这些元素中
    //最小的，也就是小顶堆y的top()
                qn.push(qx.top());
                qx.pop();
            }
        }
        r=q+1;//已经处理了前q个
        printf("%d\n",qn.top());
    //此处也是精髓：把y的top给x,保证了x中始终含有前q个元素中前i个最小元素
        qx.push(qn.top());qn.pop();
    }
    return 0;
}