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 2Sample 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\n",size[c[pos[S]][0]]-1);break; 103 case 'Q':printf("%d\n",v[find(rt,S+1)]);break; 104 } 105 } 106 return 0; 107 }