BZOJ 3223 Tyvj 1729 文艺平衡树
3223: Tyvj 1729 文艺平衡树
Description
您需要写一种数据结构(可参考题目标题),来维护一个有序数列,其中需要提供以下操作:翻转一个区间,例如原有序序列是5 4 3 2 1,翻转区间是[2,4]的话,结果是5 2 3 4 1
Input
第一行为n,m n表示初始序列有n个数,这个序列依次是(1,2……n-1,n) m表示翻转操作次数
接下来m行每行两个数[l,r] 数据保证 1<=l<=r<=n
Output
输出一行n个数字,表示原始序列经过m次变换后的结果
Sample Input
1 3
1 3
1 4
Sample Output
HINT
N,M<=100000
为了调出一个正常的splay,实在是调了太久太久。
不过至少调出来了嘛!☺
就是最普通的平衡树,而且此题还不用insert和remove。所以,就是build,splay,merge和split,再加上rev标记和pushdown。splay的cmp与treap的cmp不同,treap的cmp是按值的,而splay是按siz的(因为是下标标号)。所以说,insert的时候有k还有x(x似乎无用)。额,似乎很显然。而进入ch[1]的时候,一定不能忘了先把k减下去。
关于splay操作,按理是双旋的,但我写了随机比较单旋和双旋,似乎没有什么大区别。事实似乎也是这样。
splay单旋 | 2776 kb | 2184 ms | 2351 B |
splay双旋 | 2776 kb | 2004 ms | 2559 B |
但是,确实有理论证明。双旋可以有效地“本质化地”优化树的形态。
关于split和merge,刘汝佳建议,使用虚拟节点,但本人嫌麻烦,只需要再特判一下。因为split是把树o的前k个分入left,之后的分入right,当k为0时就不用旋也不能旋了。而且merge是把left变成“右空树”再连上 right,当left为null时就不用旋也不能旋了。第一次感觉null这么方便。
而rev标记,是只要访问就该pushdown了,无论是insert,remove,还是splay,print,如果忘了会出大事。
而maintain(update)呢?只要树形态变化就应该做。也就是build,insert,remove,rotate,还有merge和split。注意splay内部并不需要maintain,因为splay本身不过是一堆rotate。而remove时,maintain需要注意判断null,不然一旦乱搞,这棵树就完了。而null时,一般都是o->ch[0]==null&&o->ch[1]==null,在d==-1时if(o->ch[0]==null) o=o->ch[1];这样是很明显的。
我们在进行区间操作时,一般都会把操作的区间split出来,防止其他元素纠结,之后在merge回去。此题无需关心insert,但是如果有insert,插入之后还应splay一下,只有这样才能摆脱链的诅咒。
第一次敲splay,真的很苦啊。
1 /************************************************************** 2 Problem: 3223 3 User: Doggu 4 Language: C++ 5 Result: Accepted 6 Time:2184 ms 7 Memory:2776 kb 8 ****************************************************************/ 9 10 #include <cstdio> 11 #include <algorithm> 12 const int N = 100010; 13 struct Node { 14 int val, siz; 15 bool rev; 16 Node *ch[2]; 17 int cmp(int k) { 18 if(k<=ch[0]->siz) return 0; 19 if(k<=ch[0]->siz+1) return -1; 20 return 1; 21 } 22 void pushdown() {if(rev) { 23 std::swap(ch[0],ch[1]); 24 ch[0]->rev^=1; 25 ch[1]->rev^=1; 26 rev = 0; 27 }} 28 void maintain() { 29 siz=ch[0]->siz+ch[1]->siz+1; 30 } 31 }pool[N], *tail=pool, *null=pool, *root; 32 //int aa[N]; 33 void rotate(Node *&o,int d) { 34 Node *k=o->ch[d^1];o->ch[d^1]=k->ch[d];k->ch[d]=o; 35 o->maintain();k->maintain();o=k; 36 } 37 Node *build(int lf,int rg) { 38 Node *o=++tail; 39 int mid=(lf+rg)>>1; 40 o->val=mid;//aa[mid]; 41 o->ch[0]=(lf<=mid-1)?build(lf,mid-1):null; 42 o->ch[1]=(mid+1<=rg)?build(mid+1,rg):null; 43 o->maintain(); 44 return o; 45 } 46 void splay(Node *&o,int k) { 47 o->pushdown(); 48 int d=o->cmp(k); 49 if(d==1) k-=o->ch[0]->siz+1; 50 if(d!=-1) splay(o->ch[d],k), rotate(o,d^1); 51 } 52 Node *merge(Node *left,Node *right) { 53 if(left==null) return right; 54 splay(left,left->siz); 55 left->ch[1]=right;left->maintain(); 56 return left; 57 } 58 void split(Node *o,int k,Node *&left,Node *&right) { 59 if(k==0) left=null, right=o; 60 else { 61 splay(o,k); 62 left=o;right=o->ch[1]; 63 o->ch[1]=null;left->maintain(); 64 } 65 } 66 void print(Node *o) { 67 if(o==null) return ; 68 o->pushdown(); 69 print(o->ch[0]); 70 printf("%d ",o->val); 71 print(o->ch[1]); 72 } 73 int main() { 74 null->ch[0]=null->ch[1]=null; 75 int n, q, lf, rg; 76 scanf("%d%d",&n,&q); 77 //for( int i = 1; i <= n; i++ ) scanf("%d",&aa[i]); 78 root=build(1,n); 79 //printf("ROOT_SEQ:");print(root);printf("\n"); 80 while(q--) { 81 scanf("%d%d",&lf,&rg); 82 Node *left, *mid, *right, *o; 83 split(root,rg,mid,right); 84 split(mid,lf-1,left,mid); 85 //printf("LEFT_SEQ:");print(left);printf("\n"); 86 //printf("MID_SEQ:");print(mid);printf("\n"); 87 //printf("RIGHT_SEQ:");print(right);printf("\n"); 88 mid->rev^=1; 89 root=merge(merge(left,mid),right); 90 //printf("ROOT_SEQ:");print(root);printf("\n"); 91 } 92 print(root);printf("\n"); 93 return 0; 94 } 95
1 /************************************************************** 2 Problem: 3223 3 User: Doggu 4 Language: C++ 5 Result: Accepted 6 Time:2004 ms 7 Memory:2776 kb 8 ****************************************************************/ 9 10 #include <cstdio> 11 #include <algorithm> 12 const int N = 100010; 13 struct Node { 14 int val, siz; 15 bool rev; 16 Node *ch[2]; 17 int cmp(int k) { 18 if(k<=ch[0]->siz) return 0; 19 if(k<=ch[0]->siz+1) return -1; 20 return 1; 21 } 22 void pushdown() {if(rev) { 23 std::swap(ch[0],ch[1]); 24 ch[0]->rev^=1; 25 ch[1]->rev^=1; 26 rev = 0; 27 }} 28 void maintain() { 29 siz=ch[0]->siz+ch[1]->siz+1; 30 } 31 }pool[N], *tail=pool, *null=pool, *root; 32 //int aa[N]; 33 void rotate(Node *&o,int d) { 34 Node *k=o->ch[d^1];o->ch[d^1]=k->ch[d];k->ch[d]=o; 35 o->maintain();k->maintain();o=k; 36 } 37 Node *build(int lf,int rg) { 38 Node *o=++tail; 39 int mid=(lf+rg)>>1; 40 o->val=mid;//aa[mid]; 41 o->ch[0]=(lf<=mid-1)?build(lf,mid-1):null; 42 o->ch[1]=(mid+1<=rg)?build(mid+1,rg):null; 43 o->maintain(); 44 return o; 45 } 46 void splay(Node *&o,int k) { 47 o->pushdown(); 48 int d=o->cmp(k); 49 if(d==1) k-=o->ch[0]->siz+1; 50 if(d!=-1) { 51 Node *p=o->ch[d]; 52 p->pushdown(); 53 int d2=p->cmp(k); 54 int k2=(d2==0?k:k-p->ch[0]->siz-1); 55 if(d2!=-1) { 56 splay(p->ch[d2],k2); 57 if(d==d2) rotate(o,d^1);else rotate(o->ch[d],d); 58 } 59 rotate(o,d^1); 60 } 61 } 62 Node *merge(Node *left,Node *right) { 63 if(left==null) return right; 64 splay(left,left->siz); 65 left->ch[1]=right;left->maintain(); 66 return left; 67 } 68 void split(Node *o,int k,Node *&left,Node *&right) { 69 if(k==0) left=null, right=o; 70 else { 71 splay(o,k); 72 left=o;right=o->ch[1]; 73 o->ch[1]=null;left->maintain(); 74 } 75 } 76 void print(Node *o) { 77 if(o==null) return ; 78 o->pushdown(); 79 print(o->ch[0]); 80 printf("%d ",o->val); 81 print(o->ch[1]); 82 } 83 int main() { 84 null->ch[0]=null->ch[1]=null; 85 int n, q, lf, rg; 86 scanf("%d%d",&n,&q); 87 //for( int i = 1; i <= n; i++ ) scanf("%d",&aa[i]); 88 root=build(1,n); 89 //printf("ROOT_SEQ:");print(root);printf("\n"); 90 while(q--) { 91 scanf("%d%d",&lf,&rg); 92 Node *left, *mid, *right, *o; 93 split(root,rg,mid,right); 94 split(mid,lf-1,left,mid); 95 //printf("LEFT_SEQ:");print(left);printf("\n"); 96 //printf("MID_SEQ:");print(mid);printf("\n"); 97 //printf("RIGHT_SEQ:");print(right);printf("\n"); 98 mid->rev^=1; 99 root=merge(merge(left,mid),right); 100 //printf("ROOT_SEQ:");print(root);printf("\n"); 101 } 102 print(root);printf("\n"); 103 return 0; 104 } 105