BZOJ 1861: [Zjoi2006]Book 书架 splay

1861: [Zjoi2006]Book 书架

Description

小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

HINT

数据范围

100%的数据，n,m < = 80000

 

 

Source

题解：

　　splay模板题练习

　　代码能力太弱

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<cstdio>
using namespace std;
const int N = 2e5 + , inf = 2e9;

int size[N],fa[N],c[N][],n,m,sz;
int pos[N],a[N],v[N],rt;
void push_up(int k) {size[k] = size[c[k][]] + size[c[k][]] + ;}
inline 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;
        push_up(y);
        push_up(x);
}
inline void splay(int x,int &k) {
        while(x != k) {
            int y = fa[x], z = fa[y];
            if(y != k) {
                if(c[z][] == y ^ c[y][] == x) rotate(x,k);
                else rotate(y,k);
            }
            rotate(x,k);
        }
}
int find(int &k,int rank) {
        int l = c[k][], r = c[k][];
        if(size[l] +  == rank) return k;
        if(size[l] >= rank) return find(l,rank);
        else return find(r,rank - size[l] - );
}
void change(int S,int T){
         if(T == ) return;
         int p = pos[S];
         splay(p,rt);
         int rank = size[c[p][]];
         int x = find(rt,size[c[rt][]]),y = find(rt,size[c[rt][]]+);
         splay(x,rt), splay(y,c[x][]);
         c[y][] = ;size[c[y][]] = ;fa[c[y][]]=;push_up(y);push_up(x);
         if(T == inf) x = find(rt,), y = find(rt,);
         else if(T == -inf) x = find(rt,n),y = find(rt,n+);
         else {
            if(T == ) x = find(rt,rank+), y = find(rt,rank+);
            else x = find(rt,rank-), y = find  (rt,rank);
         }
         splay(x,rt), splay(y,c[x][]);
         c[y][] = p;
         fa[p] = y;
         size[p] = ;
         push_up(y),push_up(x);
}
void ins(int &k,int x,int f) {
        if(k == ) {
            k = ++sz;
            size[k] = ;
            v[k] = x;
            fa[k] = f;
            return ;
        }
        if(x < v[k]) ins(c[k][],x,k);
        else ins(c[k][],x,k);
        push_up(k);
}
int main() {
        scanf("%d%d",&n,&m);
        for(int i = ; i <= n+; ++i) scanf("%d",&a[i]), pos[a[i]] = i;
        for(int i = ; i <= n + ; ++i) ins(rt,i,),splay(sz,rt);
        while(m--) {
            char ch[];int S,T;
            scanf("%s%d",ch,&S);
            if(ch[] == 'T') change(S,inf);//,cout<<c[c[rt][1]][0]<<endl;
            if(ch[] == 'B') change(S,-inf);
            if(ch[] == 'I') scanf("%d",&T),change(S,T);
            if(ch[] == 'A') splay(pos[S],rt),printf("%d\n",size[c[rt][]] - );
            if(ch[] == 'Q') printf("%d\n",a[find(rt,S+)]);
        }
        return ;
}