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)个元素被放进了Blaek 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
思路:创建两个堆,分别为大根堆和小根堆,然后使得大根堆里的数的个数保持为k,并且大根堆的所有元素都是小于小根堆的,这样每次取大根堆的堆顶就是答案了~
#include<bits/stdc++.h>
#define maxn 400000
using namespace std;
int n,m;
int a[maxn],u[maxn];
struct cmp
{
bool operator () (const int a,const int b) const
{
return a<b;
}
};
priority_queue<int ,vector<int>,greater<int> >q;
priority_queue<int ,vector<int>,cmp >que;
int main()
{
ios::sync_with_stdio(false);
cin>>m>>n;
for(int i=1;i<=m;i++)
cin>>a[i];
for(int i=1;i<=n;i++)
cin>>u[i];
int k=1;//k标记的是在u数组中的指针的地址 ,也是get第几位
for(int i=1;i<=m;i++)
{
que.push(a[i]);
if(que.size()>0)
{
q.push(que.top());
que.pop();
}
while(u[k]==i)//通过这句使得que大根堆中的数始终保持k-1个
{
cout<<q.top()<<endl;
que.push(q.top());
q.pop();
k++;
}
}
return 0;
}