【P1806】黑匣子_NOI导刊2010提高（06） splay

题目描述

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

Input：  

7 4
3 1 -4 2 8 -1000 2

1 2 6 6

Output：

3 3 1 2

题解：

    查询可能会有点奇怪，每次在一个位置查询就把那个位置的计数器加一，再设一个now表示当前查到第now大的元素，剩下的就是平衡树的基本操作了，这里写的是splay

代码：

#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
const int MAXN=1e7;
int ch[MAXN][2],key[MAXN],cnt[MAXN],siz[MAXN],f[MAXN],add[MAXN],q[MAXN];
int n,m,x,root,sz;
using namespace std;
int getson(int x){return ch[f[x]][1]==x;}
void update(int x){siz[x]=cnt[x]+siz[ch[x][0]]+siz[ch[x][1]];}
void create(int v){key[++sz]=v;siz[sz]=cnt[sz]=1;ch[sz][0]=ch[sz][1]=f[sz]=0;}
void rotate(int x)
{
	int fa=f[x],fafa=f[fa],k=getson(x);
	ch[fa][k]=ch[x][k^1],f[ch[fa][k]]=fa;
	ch[x][k^1]=fa;f[fa]=x;
	f[x]=fafa;
	if(fafa) ch[fafa][ch[fafa][1]==fa] =x;
	update(fa);update(x);
}
void splay(int x)
{
	for(int fa;fa=f[x];rotate(x))
		if(f[fa]) rotate(getson(x)==getson(fa) ? fa : x);
	root=x;
}
void insert(int x)
{
	if(!root)
		create(x),root=sz;
	else
	{
		int now=root,fa=0;
		while(1)
		{
			if(key[now]==x){cnt[now]++;update(now);update(fa);splay(now);return ;}
			fa=now;now=ch[fa][x>key[fa]];
			if(!now)
			{
				create(x);f[sz]=fa;ch[fa][x>key[fa]]=sz;
				update(fa);splay(sz);
				return ;
			}
		}
	}
}
int findx(int x)
{
	int now=root;
	while(1)
	{
		if(x<=siz[ch[now][0]]) now=ch[now][0];
		else
		{
			int temp=siz[ch[now][0]]+cnt[now];
			x-=temp;
			if(x<=0) return now;
			now=ch[now][1];
		}
	}
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++) scanf("%d",&add[i]);
	for(int i=1;i<=m;i++) scanf("%d",&x),q[x]++;
	int now=1;
	for(int i=1;i<=n;i++)
	{
		insert(add[i]);
		while(q[i]--)
		{
			printf("%d\n",key[findx(now)]);
			now++;
		}
	}
	return 0;
}