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

输入输出样例

输入样例#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。

主席树经典好题。。。

（虽然可以用平衡树，但是代码忒长，不想写了。。。）

套模板。。。

附代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#define MAXN 200010
using namespace std;
int n,m,k,c=1;
int b[MAXN],num[MAXN],root[MAXN];
struct node{
	int l,r,sum;
}a[MAXN*20];
inline int read(){
	int date=0,w=1;char c=0;
	while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
	while(c>='0'&&c<='9'){date=date*10+c-'0';c=getchar();}
	return date*w;
}
void pushup(int rt){
	a[rt].sum=a[a[rt].l].sum+a[a[rt].r].sum;
}
void insert(int k,int l,int r,int &rt){
	int mid;
	a[c++]=a[rt];rt=c-1;
	if(l==k&&k==r){
		a[rt].sum++;
		return;
	}
	mid=l+r>>1;
	if(k<=mid)insert(k,l,mid,a[rt].l);
	else insert(k,mid+1,r,a[rt].r);
	pushup(rt);
}
int query(int i,int j,int k,int l,int r){
	if(l==r)return l;
	int mid=l+r>>1,t=a[a[j].l].sum-a[a[i].l].sum;
	if(k<=t)return query(a[i].l,a[j].l,k,l,mid);
	else return query(a[i].r,a[j].r,k-t,mid+1,r);
}
int main(){
	int x,t=0;
	n=read();m=read();
	a[0].l=a[0].r=a[0].sum=0;
	root[0]=0;
	for(int i=1;i<=n;i++){
		b[i]=read();
		num[i]=b[i];
	}
	sort(num+1,num+n+1);
	k=unique(num+1,num+n+1)-num-1;
	for(int i=1;i<=n;i++){
		x=lower_bound(num+1,num+k+1,b[i])-num;
		root[i]=root[i-1];
		insert(x,1,k,root[i]);
	}
	for(int i=1;i<=m;i++){
		t++;x=read();
		printf("%d\n",num[query(root[0],root[x],t,1,k)]);
	}
	return 0;
}