bzoj1861 [Zjoi2006]Book 书架 splay

小T有一个很大的书柜。这个书柜的构造有些独特，即书柜里的书是从上至下堆放成一列。她用1到n的正整数给每本书都编了号。 小T在看书的时候，每次取出一本书，看完后放回书柜然后再拿下一本。由于这些书太有吸引力了，所以她看完后常常会忘记原来是放在书柜的什么位置。不过小T的记忆力是非常好的，所以每次放书的时候至少能够将那本书放在拿出来时的位置附近，比如说她拿的时候这本书上面有X本书，那么放回去时这本书上面就只可能有X-1、X或X+1本书。 当然也有特殊情况，比如在看书的时候突然电话响了或者有朋友来访。这时候粗心的小T会随手把书放在书柜里所有书的最上面或者最下面，然后转身离开。 久而久之，小T的书柜里的书的顺序就会越来越乱，找到特定的编号的书就变得越来越困难。于是她想请你帮她编写一个图书管理程序，处理她看书时的一些操作，以及回答她的两个提问：(1)编号为X的书在书柜的什么位置；(2)从上到下第i本书的编号是多少。

Input

第一行有两个数n，m，分别表示书的个数以及命令的条数；第二行为n个正整数：第i个数表示初始时从上至下第i个位置放置的书的编号；第三行到m+2行，每行一条命令。命令有5种形式： 1． Top S——表示把编号为S的书房在最上面。 2． Bottom S——表示把编号为S的书房在最下面。 3． Insert S T——T∈{-1，0，1}，若编号为S的书上面有X本书，则这条命令表示把这本书放回去后它的上面有X+T本书； 4． Ask S——询问编号为S的书的上面目前有多少本书。 5． Query S——询问从上面数起的第S本书的编号。

Output

对于每一条Ask或Query语句你应该输出一行，一个数，代表询问的答案。

Sample Input

10 10
1 3 2 7 5 8 10 4 9 6
Query 3
Top 5
Ask 6
Bottom 3
Ask 3
Top 6
Insert 4 –1
Query 5
Query 2
Ask 2

Sample Output

2
9
9
7
5
3
数据范围
30%的数据，n,m < = 10000
100%的数据，n,m < = 80000

这是一道splay的模板题，算是入门了吧，毕竟不会所有平衡树，过会得开一讲了，

这道题就是裸的splay，真的比较水，整理的是hzwer大佬的模板，注释在代码上来。

 #include<cstdio>
 #include<iostream>
 #include<algorithm>
 #include<cmath>
 #include<cstring>

 using namespace std;

 const int NN=,inf=1e9+;

 int n,m,rt,sz;
 int c[NN][],fa[NN],deep[NN];
 int a[NN],size[NN],v[NN],pos[NN];

 void update(int k)//仅仅用来得到k的大小
 {
     size[k]=size[c[k][]]+size[c[k][]]+;
 }
 void rotate(int x,int &k)//一次旋转
 {
     int y=fa[x],z=fa[y],l,r;
     if (c[y][]==x) l=;
     else l=;
     r=l^;
     if (y==k) k=x;
     else if (c[z][]==y) c[z][]=x;
          else c[z][]=x;
     fa[x]=z,fa[y]=x,fa[c[x][r]]=y;
     c[y][l]=c[x][r],c[x][r]=y;
     update(y),update(x);//普通的一次操作。
 }
 void splay(int x,int &k)
 {
     while(x!=k)
     {
         int y=fa[x],z=fa[y];
         if (y!=k)
         {
             if (c[y][]==x^c[z][]==y)//这个是判断zig还是zag，来决定谁先交换。
                 rotate(x,k);
             else rotate(y,k);
         }
         rotate(x,k);
     }
 }
 void build(int l,int r,int f)//l,r是表示区间，f表示根
 {
     if (l>r) return;
     int now=l,last=f;
     if (l==r)
     {
         v[now]=a[l],size[now]=,fa[now]=last;
         if (l<f) c[last][]=now;//右边左边的问题
         else c[last][]=now;
         return;
     }
     int mid=(l+r)>>;
     now=mid;
     build(l,mid-,mid);build(mid+,r,mid);
     v[now]=a[mid],fa[now]=last;
     update(now);
     if (mid<f) c[last][]=now;
     else c[last][]=now;
 }
 int find(int k,int rank)//以k为根的树中找rank
 {
     int l=c[k][],r=c[k][];
     if (size[l]+==rank) return k;
     else if (size[l]>=rank) return find(l,rank);
          else return find(r,rank-size[l]-);
 }
 void del(int k)
 {
     int x,y,z;
     x=find(rt,k-),y=find(rt,k+);
     splay(x,rt),splay(y,c[x][]);
     z=c[y][],c[y][]=,fa[z]=size[z]=;//把前驱转到根，把后驱转到根右子树，然后根右子树左儿子就是要删除的值。
     update(y),update(x);
 }
 void move(int k,int val)//就是将编号为k的放到val值的地方。
 {
     int x,y,z=pos[k],rank;
     splay(z,rt);
     rank=size[c[z][]]+;
     del(rank);
     if (val==inf) x=find(rt,n),y=find(rt,n+);
     else if (val==-inf) x=find(rt,),y=find(rt,);
          else x=find(rt,rank+val-),y=find(rt,rank+val);
     splay(x,rt),splay(y,c[x][]);
     size[z]=,fa[z]=y,c[y][]=z;
     update(y),update(x);
 }
 int main()
 {
     scanf("%d%d",&n,&m);
     for (int i=;i<=n+;i++)//特别把1空出来，防止越界。
     {
         scanf("%d",&a[i]);
         pos[a[i]]=i;//反向映射。
     }
     build(,n+,);
     rt=(+n)>>;
     char ch[];int S,T;
     for (int i=;i<=m;i++)
     {
         scanf("%s%d",ch,&S);
         switch(ch[])
         {
             case'T':move(S,-inf);break;
             case'B':move(S,inf);break;
             case'I':scanf("%d",&T);move(S,T);break;
             case'A':splay(pos[S],rt);printf("%d\n",size[c[pos[S]][]]-);break;
             case'Q':printf("%d\n",v[find(rt,S+)]);break;
         }
     }
 }