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
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
9
9
7
5
3
HINT
数据范围
这个题实际是在让我们维护一个序列,支持每次将其中的一个值放到第一个或者最后一个或者和前面的交换或者和后面的交换,询问第i个的值,和值为x的在第几位,
光看询问是平衡树的操作,看到序列想到splay,我们可以按照下标建立splay,如果放到第一个不就相当于把原来的序列头转到根,然后成为x的右儿子,
放到最后一个就是x的左儿子,所以放到最后和第一个解决了,向前交换和向后交换,设交换之后x的位置是pos,那么也就是把pos-1作为左儿子,pos+1作为右儿子,只要删除x,把pos-1转到根,再把pos+1转到pos-1的右儿子,即可。。。
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
1 #include<iostream> 2 #include<cstdlib> 3 #include<cmath> 4 #include<cstring> 5 #include<cstdio> 6 #include<algorithm> 7 #include<string> 8 #include<map> 9 #include<queue> 10 #include<vector> 11 #include<set> 12 #define inf 1000000000 13 #define maxn 80000+5 14 #define maxm 10000+5 15 #define eps 1e-10 16 #define ll long long 17 #define for0(i,n) for(int i=0;i<=(n);i++) 18 #define for1(i,n) for(int i=1;i<=(n);i++) 19 #define for2(i,x,y) for(int i=(x);i<=(y);i++) 20 #define for3(i,x,y) for(int i=(x);i>=(y);i--) 21 #define for4(i,x) for(int i=head[x],y=e[i].go;i;i=e[i].next,y=e[i].go) 22 using namespace std; 23 int read(){ 24 int x=0,f=1;char ch=getchar(); 25 while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} 26 while(ch>='0'&&ch<='9'){x=10*x+ch-'0';ch=getchar();} 27 return x*f; 28 } 29 int n,m,sz,rt; 30 int fa[maxn],c[maxn][2],deep[maxn]; 31 int s[maxn],a[maxn],v[maxn],pos[maxn]; 32 void pushup(int x){ 33 int l=c[x][0],r=c[x][1]; 34 s[x]=s[l]+s[r]+1; 35 } 36 void rotate(int x,int &k){ 37 int y=fa[x],z=fa[y],l=(c[y][1]==x),r=l^1; 38 if(y==k)k=x; 39 else c[z][(c[z][1]==y)]=x; 40 fa[x]=z;fa[y]=x;fa[c[x][r]]=y; 41 c[y][l]=c[x][r];c[x][r]=y; 42 pushup(y);pushup(x); 43 } 44 void splay(int x,int &k){ 45 while(x!=k){ 46 int y=fa[x],z=fa[y]; 47 if(y!=k){ 48 if(c[y][0]==x^c[z][0]==y)rotate(x,k); 49 else rotate(x,k); 50 } 51 else rotate(x,k); 52 } 53 } 54 int find(int k,int rk){ 55 int l=c[k][0],r=c[k][1]; 56 if(s[l]+1==rk)return k; 57 else if(s[l]>=rk)return find(l,rk); 58 else return find(r,rk-s[l]-1); 59 } 60 void del(int k){ 61 int x,y,z; 62 x=find(rt,k-1);y=find(rt,k+1); 63 splay(x,rt);splay(y,c[x][1]); 64 z=c[y][0];c[y][0]=0;fa[z]=s[z]=0; 65 pushup(y);pushup(x); 66 } 67 void move(int k,int val){ 68 int x,y,z=pos[k],rank; 69 splay(z,rt);rank=s[c[z][0]]+1; 70 del(rank); 71 if(val==inf)x=find(rt,n),y=find(rt,n+1); 72 else if(val==-inf)x=find(rt,1),y=find(rt,2); 73 else x=find(rt,rank+val-1),y=find(rt,rank+val); 74 splay(x,rt);splay(y,c[x][1]); 75 s[z]=1;fa[z]=y;c[y][0]=z; 76 pushup(y);pushup(x); 77 } 78 void build(int l,int r,int f){ 79 if(l>r)return ; 80 int now=l,last=f; 81 if(l==r){ 82 v[now]=a[l];fa[now]=last;s[now]=1; 83 if(l<f)c[last][0]=now; 84 else c[last][1]=now; 85 return ; 86 } 87 int mid=(l+r)>>1;now=mid; 88 build(l,mid-1,mid);build(mid+1,r,mid); 89 v[now]=a[mid];fa[now]=last;pushup(mid); 90 if(mid<=f)c[last][0]=now; 91 else c[last][1]=now; 92 } 93 int main(){ 94 //freopen("input.txt","r",stdin); 95 //freopen("output.txt","w",stdout); 96 n=read();m=read(); 97 for(int i=2;i<=n+1;i++) 98 a[i]=read(),pos[a[i]]=i; 99 build(1,n+2,0); 100 rt=(3+n)>>1; 101 int S,T; 102 for1(i,m){ 103 char ch=getchar(); 104 while(ch!='T'&&ch!='B'&&ch!='I'&&ch!='A'&&ch!='Q')ch=getchar(); 105 S=read(); 106 switch(ch){ 107 case 'T':move(S,-inf);break; 108 case 'B':move(S,inf);break; 109 case 'I':T=read();move(S,T);break; 110 case 'A':splay(pos[S],rt);printf("%d\n",s[c[pos[S]][0]]-1);break; 111 case 'Q':printf("%d\n",v[find(rt,S+1)]);break; 112 } 113 } 114 return 0; 115 }