BZOJ 1861书架

小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 Input10 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 Output2 9 9 7 5 3Hint

数据范围

100%的数据,n,m < = 80000
 题解:这道题就是Splay模板题TVT
 
参考代码:
  1 #include<bits/stdc++.h>
  2 using namespace std;
  3 const int inf=0x3f3f3f3f;
  4 inline int read()
  5 {
  6     int x=0,f=1;char ch=getchar();
  7     while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
  8     while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
  9     return x*f;
 10 }
 11 int n,m,rt,sz;
 12 int c[80005][2],fa[80005],deep[80005];
 13 int a[80005],size[80005],v[80005],pos[80005];//a[i]:表示第i个位置是第几本书,pos[i]:表示第i本书在哪个位置 
 14 void update(int k)
 15 {
 16     size[k]=size[c[k][0]]+size[c[k][1]]+1;
 17 }
 18 void rotate(int x,int &k)
 19 {
 20     int y=fa[x],z=fa[y],l,r;
 21     if(c[y][0]==x)l=0;else l=1;r=l^1;
 22     if(y==k)k=x;
 23     else {if(c[z][0]==y)c[z][0]=x;else c[z][1]=x;}
 24     fa[x]=z;fa[y]=x;fa[c[x][r]]=y;
 25     c[y][l]=c[x][r];c[x][r]=y;
 26     update(y);update(x);
 27 }
 28 void splay(int x,int &k)
 29 {
 30     while(x!=k)
 31     {
 32         int y=fa[x],z=fa[y];
 33         if(y!=k)
 34         {    
 35             if(c[y][0]==x^c[z][0]==y)
 36                 rotate(x,k);
 37             else rotate(y,k);
 38         }
 39         rotate(x,k);
 40     }
 41 }
 42 void build(int l,int r,int f)
 43 {
 44     if(l>r)return;
 45     int now=l,last=f;
 46     if(l==r)
 47     {
 48         v[now]=a[l];size[now]=1;fa[now]=last;
 49         if(l<f) c[last][0]=now;
 50         else c[last][1]=now;
 51         return;
 52     }
 53     int mid=(l+r)>>1;now=mid;
 54     build(l,mid-1,mid);build(mid+1,r,mid);
 55     v[now]=a[mid];fa[now]=last;update(now);
 56     if(mid<f) c[last][0]=now;
 57     else c[last][1]=now;
 58 }
 59 int find(int k,int rank)
 60 {
 61     int l=c[k][0],r=c[k][1];
 62     if(size[l]+1==rank)return k;
 63     else if(size[l]>=rank)return find(l,rank);
 64     else return find(r,rank-size[l]-1);
 65 }
 66 void del(int k)
 67 {
 68     int x,y,z;
 69     x=find(rt,k-1);y=find(rt,k+1);
 70     splay(x,rt);splay(y,c[x][1]);
 71     z=c[y][0];c[y][0]=0;fa[z]=size[z]=0;
 72     update(y);update(x);
 73 }
 74 void move(int k,int val)
 75 {
 76     int x,y,z=pos[k],rank;
 77     splay(z,rt);rank=size[c[z][0]]+1;
 78     del(rank);
 79     if(val==inf) x=find(rt,n),y=find(rt,n+1);//
 80     else if(val==-inf) x=find(rt,1),y=find(rt,2);//
 81     else x=find(rt,rank+val-1),y=find(rt,rank+val);
 82     splay(x,rt);splay(y,c[x][1]);
 83     size[z]=1;fa[z]=y;c[y][0]=z;
 84     update(y);update(x);
 85 }
 86 int main()
 87 {
 88     n=read();m=read();
 89     for(int i=2;i<=n+1;i++)
 90         a[i]=read(),pos[a[i]]=i;
 91     build(1,n+2,0);
 92     rt=(3+n)>>1;
 93     char ch[10];int S,T;
 94     for(int i=1;i<=m;i++)
 95     {
 96         scanf("%s",ch);S=read();
 97         switch(ch[0])
 98         {
 99         case 'T':move(S,-inf);break;
100         case 'B':move(S,inf);break;
101         case 'I':T=read();move(S,T);break;
102         case 'A':splay(pos[S],rt);printf("%d
",size[c[pos[S]][0]]-1);break;
103         case 'Q':printf("%d
",v[find(rt,S+1)]);break;
104         }
105     }
106     return 0;
107 }
View Code
原文地址:https://www.cnblogs.com/csushl/p/10314246.html