题目描述
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根据命令串所得出的输出串,一个数字一行。
输入输出样例
输入样例#1: 复制
7 4 3 1 -4 2 8 -1000 2 1 2 6 6
输出样例#1: 复制
3 3 1 2
说明
对于30%的数据,M≤10000;
对于50%的数据,M≤100000:
对于100%的数据,M≤200000。
题解:对于堆来说,只能求出当前序列最小的,那么对于这道题我们可以维护两个堆,一个大顶堆和一个小顶堆,大顶堆里面维护i个最小的元素,小顶堆则维护剩下的元素,那么此时小顶堆的堆顶元素就是第i+1小的元素,对于每次添加元素的时候,先考虑是否大于大顶堆的堆顶元素,若大于则将其加入小顶堆,否则就将大顶堆的堆顶元素弹出到小顶堆,再加入大顶堆中。输出则直接输出小顶堆的堆顶元素,然后将堆顶元素加入大顶堆即可。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e6+7;
void read(int &x){scanf("%d",&x);}
int a[maxn];
int main()
{
int n;
scanf("%d",&n);
priority_queue<int,vector<int>,less<int> >q1;
priority_queue<int,vector<int>,greater<int> >q2;
int t;
read(t);
int len=0;
for(int i=1;i<=n;i++)
read(a[i]);
int st=1;
for(int i=1;i<=t;i++)
{
int ti;
read(ti);
for(;st<=ti;st++)
{
if(i==1)
q2.push(a[st]);
else
{
if(a[st]<q1.top())
{
q2.push(q1.top());
q1.pop();
q1.push(a[st]);
}
else
q2.push(a[st]);
}
}
printf("%d\n",q2.top());
q1.push(q2.top());
q2.pop();
}
return 0;
}