问题引入:黑匣子
Black Box是一种原始的数据库。它可以储存一个整数数组,还有一个特别的变量i。最开始的时候Black Box是空的.而i等于0。这个Black Box要处理一串命令。
命令只有两种:
ADD(x):把x元素放进BlackBox;
GET:i加1,然后输出Blackhox中第i小的数。
记住:第i小的数,就是Black Box里的数的按从小到大的顺序排序后的第i个元素。
现在要求找出对于给定的命令串的最好的处理方法。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根据命令串所得出的输出串,一个数字一行。
这道题来源于2010年的NOI导刊。看到这题的感受如何?不就是排序吗?有道理,我们来排排看!
假设我们用O(n log n)的算法……最多应付……3000数据……
那么,这种慢慢插入的方法,是不是能令人想到堆呢?没错,其实就是一个不断进行堆操作的过程。
不过,想到这一步,还是眼泪掉下来……
堆的性质不能保证数列有序啊!
但是仔细想想,这题参与操作的无非就是恰在某条分割线上位置的数据……每一次Get就是移动这条分割线,Add就是往分割线处插数据……
突然有一个漏斗的形状浮现在脑海中!
没错……就是一个漏斗!两个堆,像漏斗一样地摆放。上面是大顶堆,下面是小顶堆!大功告成。
【事实上这题可以用PQ模版】
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
#define read(x) scanf("%d",&x)
#define write(x) printf("%d\n",x)
int a[200005],u[200005],na,nu;
vector <int>vb,vs;
int add(int x)
{
int tmp;
if(vb.empty()||x>vb[0])
vs.push_back(x),
push_heap(vs.begin(),vs.end(),greater<int>());
else
tmp=vb[0],
pop_heap(vb.begin(),vb.end()),
vb.pop_back(),
vb.push_back(x),
push_heap(vb.begin(),vb.end()),
vs.push_back(tmp),
push_heap(vs.begin(),vs.end(),greater<int>());
return 1;
}
void get()
{
int tmp=vs[0];
pop_heap(vs.begin(),vs.end(),greater<int>());
vs.pop_back();
vb.push_back(tmp);
push_heap(vb.begin(),vb.end());
write(tmp);
return;
}
int main()
{
read(na);read(nu);
int i,j,t;
for(i=1;i<=na;i++) read(a[i]);
for(i=1;i<=nu;i++) read(u[i]);
i=0;j=1;
while(++i<=na&&add(a[i]))
while(i==u[j]) get(), j++;
return 0;
}