BZOJ 1861: [Zjoi2006]Book 書架 （splay）

1861: [Zjoi2006]Book 書架

Time Limit: 4 Sec  Memory Limit: 64 MB
Submit: 1453  Solved: 822
[Submit][Status][Discuss]

Description

小T有一個很大的書櫃。這個書櫃的構造有些獨特，即書櫃裏的書是從上至下堆放成一列。她用1到n的正整數給每本書都編了號。 小T在看書的時候，每次取出一本書，看完後放回書櫃而後再拿下一本。因爲這些書太有吸引力了，因此她看完後經常會忘記原來是放在書櫃的什麼位置。不太小T的記憶力是很是好的，因此每次放書的時候至少可以將那本書放在拿出來時的位置附近，好比說她拿的時候這本書上面有X本書，那麼放回去時這本書上面就只可能有X-一、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

Day2

 

————————————————————————

splay維護序列的裸題

只須要維護一個子樹大小size

1.top 將一個點刪除，插入到根節點最左兒子

2.bottom 將一個點刪除，插入到根節點最右兒子

3.insert +1 -1 （記錄s前一個點爲prev，後一個點爲next）

-1 prev提根，s插在prev和它的左兒子之間

+1 next提根，s插在next和它的右兒子之間

4.ask  s提根，左兒子的size

5.query 遞歸查找便可

  1 #include <iostream>
  2 #include <cstdio>
  3 #include <cstring>
  4 #include <algorithm>
  5 #include <queue>
  6 #include <set>
  7 #include <vector>
  8 #include <string.h>
  9 #define siji(i,x,y) for(int i=(x);i<=(y);++i)
 10 #define gongzi(j,x,y) for(int j=(x);j>=(y);--j)
 11 #define xiaosiji(i,x,y) for(int i=(x);i<(y);++i)
 12 #define sigongzi(j,x,y) for(int j=(x);j>(y);--j)
 13 #define inf 0x1f1f1f1f
 14 #define ivorysi
 15 #define mo 97797977
 16 #define hash 974711
 17 #define base 47
 18 #define MAXN 100005
 19 #define fi first
 20 #define se second
 21 #define pii pair<int,int>
 22 using namespace std;
 23 typedef long long ll;
 24 struct node {
 25     int size,son[2],fa;    
 26     void clear() {
 27         size=son[0]=son[1]=fa=0;
 28     }
 29 }tree[MAXN];
 30 int root;
 31 //維護size
 32 void update(int x,int c) {
 33     int k=tree[x].size;
 34     int o=tree[x].fa;
 35     tree[x].size=tree[o].size;
 36     tree[o].size=tree[o].size-k+tree[tree[x].son[c]].size;
 37 }
 38 void rotate(int x,int c) {
 39     update(x,c);
 40     int o=tree[x].fa;
 41     if(tree[o].fa!=0) {
 42         int t= o==tree[tree[o].fa].son[0] ? 0 : 1;
 43         tree[tree[o].fa].son[t]=x;
 44     }
 45     tree[o].son[c^1]=tree[x].son[c];
 46     if(tree[o].son[c^1])tree[tree[o].son[c^1]].fa=o;
 47     tree[x].son[c]=o;
 48     tree[x].fa=tree[o].fa;
 49     tree[o].fa=x;
 50     
 51 }
 52 void splay(int x,int y) {
 53     while(tree[x].fa!=y) {
 54         int k1= x==tree[tree[x].fa].son[0] ? 1 : 0;
 55         int k2= tree[x].fa == tree[tree[tree[x].fa].fa].son[0] ? 1 : 0;
 56         if(tree[tree[x].fa].fa!=y && k2==k1) {
 57             rotate(tree[x].fa,k1);
 58             rotate(x,k1);
 59         }
 60         else {
 61             rotate(x,k1);
 62         }
 63     }
 64     if(y==0) root=x;
 65     else {
 66 
 67     }
 68 }
 69 int find(int x,int c) {
 70     splay(x,0);
 71     int t=tree[x].son[c];
 72     while(tree[t].son[c^1]!=0) {
 73         t=tree[t].son[c^1];
 74     }
 75     return t;
 76 }
 77 void delete_x(int prev,int next) {
 78     if(prev!=0 && next!=0) {
 79         splay(prev,0);
 80         splay(next,prev);
 81         tree[tree[root].son[1]].son[0]=0;
 82         --tree[tree[root].son[1]].size;
 83     }
 84     else if(prev!=0) {
 85         splay(prev,0);
 86         tree[root].son[1]=0;
 87     }
 88     else {
 89         splay(next,0);
 90         tree[root].son[0]=0;
 91     }
 92     --tree[root].size;
 93 }
 94 //0是top 1是bottom
 95 void in_hurry(int x,int c) {
 96     int pr=find(x,0),ne=find(x,1);
 97     delete_x(pr,ne);
 98     int t=root;
 99     ++tree[t].size;
100     tree[x].son[c^1]=tree[t].son[c];
101     tree[tree[t].son[c]].fa=x;
102     tree[x].size=tree[tree[t].son[c]].size+1;
103     tree[x].fa=t;
104     tree[t].son[c]=x;
105 }
106 int ask(int x) {
107     splay(x,0);
108     return tree[tree[x].son[0]].size;
109 }
110 void insert(int x,int c) {
111     if(c==0) return;
112     if(ask(x)==0 && c==-1) return;
113     int pr=find(x,0),ne=find(x,1);
114     delete_x(pr,ne);
115     int t= c==-1 ? pr : ne;
116     splay(t,0);
117     c= c==-1 ? 0 : 1;
118     ++tree[t].size;
119     tree[x].clear();
120     tree[x].son[c]=tree[t].son[c];
121     tree[tree[t].son[c]].fa=x;
122     tree[x].size=tree[tree[t].son[c]].size+1;
123     tree[t].son[c]=x;
124     tree[x].fa=t;
125     splay(x,0);
126 }
127 
128 void query(int x,int s) {
129     int temp=tree[tree[x].son[0]].size+1;
130     if(s==temp) {splay(x,0);return;}
131     int t=s<temp ? 0 : 1;
132     s = s>=temp ? s-temp : s; 
133     query(tree[x].son[t],s);
134 }
135 int n,m;
136 void init() {
137     scanf("%d%d",&n,&m);
138     int a=0,b;
139     siji(i,1,n) {
140         scanf("%d",&b);
141         tree[b].fa=a;
142         if(a!=0) { 
143             tree[a].son[1]=b;
144         }
145         else {root=b;}
146         tree[b].size=n-i+1;
147         a=b;
148     }
149 
150 }
151 void solve() {
152     init();
153     char ord[15];
154     int s,t;
155     siji(i,1,m) {
156         scanf("%s",ord);
157         if(ord[0]=='T' || ord[0]=='B' ) {
158             t=ord[0]=='T' ? 0 : 1;
159             scanf("%d",&s);
160             in_hurry(s,t);
161         }
162         else if(ord[0]=='I') {
163             scanf("%d%d",&s,&t);
164             insert(s,t);
165         }
166         else if(ord[0]=='A'){
167             scanf("%d",&s);
168             printf("%d\n",ask(s));
169         }
170         else if(ord[0]=='Q') {
171             scanf("%d",&s);
172             query(root,s);
173             printf("%d\n",root);
174         }
175     }
176 }
177 int main(int argc, char const *argv[])
178 {
179 #ifdef ivorysi
180     freopen("f1.in","r",stdin);
181 #endif
182     solve();
183 }