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

Posted 2020-08-29 sigongzi

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-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

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 }