P1801 黑匣子

题目描述

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

命令只有两种：

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

• GET：ii 加 11，然后输出 Black Box 中第 ii 小的数。

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

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

序号	操作	ii	数据库	输出
1	ADD(3)	00	33	/
2	GET	11	33	33
3	ADD(1)	11	1,31,3	/
4	GET	22	1,31,3	33
5	ADD(-4)	22	-4,1,3−4,1,3	/
6	ADD(2)	22	-4,1,2,3−4,1,2,3	/
7	ADD(8)	22	-4,1,2,3,8−4,1,2,3,8	/
8	ADD(-1000)	22	-1000,-4,1,2,3,8−1000,−4,1,2,3,8	/
9	GET	33	-1000,-4,1,2,3,8−1000,−4,1,2,3,8	11
10	GET	44	-1000,-4,1,2,3,8−1000,−4,1,2,3,8	22
11	ADD(2)	44	-1000,-4,1,2,2,3,8−1000,−4,1,2,2,3,8	/

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

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

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

输入格式

第一行两个整数 mm 和 nn，表示元素的个数和 GET 命令的个数。

第二行共 mm 个整数，从左至右第 ii 个整数为 a_iai​，用空格隔开。

第三行共 nn 个整数，从左至右第 ii 个整数为 u_iui​，用空格隔开。

输出格式

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

输入输出样例

输入 #1复制

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

输出 #1复制

3
3
1
2

说明/提示

数据规模与约定

• 对于 30\%30% 的数据，1 \leq n,m \leq 10^{4}1≤n,m≤104。
• 对于 50\%50% 的数据，1 \leq n,m \leq 10^{5}1≤n,m≤105。
• 对于 100\%100% 的数据，1 \leq n,m \leq 2 \times 10^{5},|a_i| \leq 2 \times 10^{9}1≤n,m≤2×105,∣ai​∣≤2×109，保证 uu 序列单调不降。

#include<bits/stdc++.h>
using namespace std;
#include<stdio.h>
priority_queue<int ,vector<int>,greater<int>> q1;//graeter 对应着上面的小顶堆
priority_queue<int ,vector<int>,less<int>> q2;//less对应着下面的大顶堆
//大顶堆是正三角，数值小，对应less
//小顶堆是倒三角， 数值大，对应greater
int a[200005];
int main(){
	int i=0;
	int m,n;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
	}
	int j=1;
	for(int i=1;i<=m;i++){//既代表get的次数，也代表get要输出的从小到大第i个值 
		int t;
		scanf("%d",&t);
		for(int k=j;k<=t;k++){
			//push 插入并排序 
			q2.push(a[k]);//输入大根堆，按从下到上依次增大的顺序排 
			if(q2.size()==i) q1.push(q2.top()),q2.pop();
			//如果大根堆排满了，将堆顶即最大的数值给小顶堆 			
		}
		j=t+1;//避免没必要的运算，加快速度 
		printf("%d\n",q1.top());//要求输出从小到大第i个
		//大顶堆一共有i-1个数值，故q2.top（）对应着从小到大第n-1个
		//q1.top（）对应着从小到大第n个 
		q2.push(q1.top());
		q1.pop(); //将小根堆堆顶数值即最小数给大顶堆，这步是为了应对相邻两次t相同的情况 
		//相邻两次t相同时，上面以k为自变量的for循环不会进行
		//这部保证了q1.top（）为从小到大第i个 
	} 
	return 0;
}