P1801 黑匣子_NOI导刊2010提高（06）

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根据命令串所得出的输出串，一个数字一行。

输入输出样例

输入

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

输出

3
3
1
2

说明/提示

对于30%的数据，M≤10000;

对于50%的数据，M≤100000：

对于100%的数据，M≤200000。

代码：

#include<bits/stdc++.h>
using namespace std;

priority_queue<long long> q1;
priority_queue<long long,vector<long long>,greater<long long> >q2;

long long a[200005];

int main()
{
	int m,n;
	cin >> m >> n;
	for(int i = 1 ; i <= m ; i++)
	{
		cin >> a[i];
	}
	int tnum = 1;
	for(int i = 1 ; i <= n ; i++)
	{
		int k;
		cin >> k;
		while(tnum<=k)
		{
			q1.push(a[tnum]);
			long long p = q1.top();
			q1.pop();
			q2.push(p);
			tnum++;
		}
		
		q1.push(q2.top());
		cout << q2.top() << endl;
		q2.pop();
	}
	
	return 0;
}

思路：

首先，我们定义一个大根堆和一个小根堆。开始把元素放进大根堆，然后把大根堆堆顶赋给小根堆，要输出时，我们输出小根堆堆顶，然后把小根堆堆顶弹出放进大根堆里面。

（1）第一次输出之前，我们已知大根堆是没有元素的，而小根堆放满了输出前的所有元素，输出时，我们输出小根堆堆顶（即第一次输出的最小数（i=1）），然后弹出小根堆堆顶（即第一次输出的最小数（i=1））放入大根堆。

（2）第二次输出之前，在第一次输出之后，我们把（i=1）的最小数放进了大根堆里面。之后每次把元素放进大根堆，然后把大根堆堆顶赋给小根堆，要输出时，我们输出小根堆堆顶，然后把小根堆堆顶弹出放进大根堆里面。其实这时，我们每放进大根堆一个数，这个数都会与大根堆的另一个数比较，最后弹出的数其实是大于大根堆还剩的一个数（即i=2）的最小数，而（i=1）的最小数会一直留在大根堆里面，而不会去到小根堆里面。所以第二次输出时，我们输出小根堆的堆顶（i=2）的最小数，然后再弹出放进大根堆里面。

（3）循环下去，每次输出的数即小根堆堆顶的数。