【题目描述】:
Black Box是一种原始的数据库。它可以储存一个整数数组,还有一个特别的变量i。最开始的时候Black Box是空的.而i等于0。这个Black Box要处理一串命令。
命令只有两种:
ADD(x):把x元素放进BlackBox;
GET:i加1,然后输出Blackhox中第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) 3 -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和GET命令分别最多200000个。现在用两个整数数组来表示命令串:
A(1),A(2),…A(M):一串将要被放进Black Box的元素。每个数都是绝对值不超过2,000,000,000的整数,M<=200,000。例如上面的例子就是A=(3,1,一4,2,8,-1000,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)。
【样例输入】:
7 4
3 1 -4 2 8 -1000 2
1 2 6 6
【样例输出】:
3
3
1
2
【时间限制、数据范围及描述】:
时间:3s 空间:128M
对于 30%的数据,M<=10,000;
对于 50%的数据,M<=100,000:
对于100%的数据,M<=200,000。
【解题思路】:
这题题面看起来很麻烦,但其实就是一个对顶堆问题。对顶堆即同时使用一个大根堆和一个小根堆维护数据。
所以对于add操作,我们先将元素放入下面的大根堆内,从堆顶不断取出元素放到小根堆直到大根堆元素个数为i,这样大根堆的根就是第i小的元素,同时可以保证对顶堆的性质
同理,对于get操作,我们先输出大根堆的根,然后将小根堆的根移到大根堆(因为i每次加一,要始终保证大根堆有i个元素),这样仍保证对顶堆的性质不变
【AC代码】:
#include<bits/stdc++.h>
#define M(a,b) memset(a,b,sizeof(a))
#define pi 3.1415926
using namespace std;
int i,j,q,n,m,r=1;
int a[200005];
priority_queue<int>A;
priority_queue<int,vector<int>,greater<int> >B;
int main(){
scanf("%d%d",&n,&m);
for(i=1;i<=n;i++)scanf("%d",&a[i]);
for(i=1;i<=m;i++){
scanf("%d",&q);
for(j=r;j<=q;j++){
A.push(a[j]);
if(A.size()==i){
B.push(A.top());
A.pop();
}
}
r=q+1;
printf("%d\n",B.top());
A.push(B.top());
B.pop();
}
return 0;
}