◆学时·VI◆ SPLAY伸展树
平衡树之多,学之不尽也……
◇算法概述
二叉排序树的一种,自动平衡,由 Tarjan 提出并实现。得名于特有的 Splay 操作。
Splay操作:将节点u通过单旋、双旋移动到某一个指定位置。
主要目的是将访问频率高的节点在不改变原顺序的前提下移动到尽量靠近根节点的位置,以此来解决同一个(相似)问题的多次查询。
但是在非降序查询每一个节点后,Splay 树会变为一条链,降低运算效率。
◇原理&细解
(1)旋转操作
二叉排序树必须满足 左儿子<根节点<右儿子 ,即使在旋转过后也是如此。因此旋转操作(Rotate)是Splay平衡树的一个重要组成部分。而在Splay操作中,旋转分单旋和双旋。
单旋:
由于Rotate分成两种情况,许多OIer直接把两种情况分类讨论写在程序里,这样就使得Rotate()函数及其之长。但是老师教了我们一个不错的俭省代码的方法(~^o^~):
首先我们定义x的左儿子为 tree[x].ch[0],右儿子为 ch[1],再在Rotate()函数的参数表中加上"d",d=1表示右旋,0表示左旋。
void Rotate(Node *x,int d) { Node *y=x->fa; y->ch[!d]=x->ch[d];x->fa=y->fa; if(x->ch[d]!=NULL) x->ch[d]->fa=y; if(y->fa!=NULL) y->fa->ch[y->fa->ch[1]==y]=x; y->fa=x;x->ch[d]=y; if(y==root) root=x; Update(y); }
完美地契合了上图的规律,从而达到简短代码的目的!
双旋:
不用怎么解释……其实就是3个点(儿子X,父亲Y,祖父Z)之间将儿子X转移到祖父Z位置的2次旋转操作。第一次旋转能够将儿子X旋转到父亲Y位置,此时的旋转和祖父Z没有关系,就看成X,Y的旋转;第一次旋转后,Y就成了X的一棵子树,所以第二次旋转是Z和X之间的……总而言之就是两次单旋,只是注意旋转方向,保证原有关系不变。
举个例子:
reader 们可以把剩下的3种自己试一试,有什么不懂的可以在文末的邮箱处ask我 (^.^)~
(2)SPLAY操作
实质是旋转的组合……
作为Splay树的核心,它能够实现将指定节点旋转到某一个位置(或某一个节点的儿子的位置)的操作。通过Splay操作,我们可以每一次将查询的节点向高处提,从而下一次访问该节点时速度加快。
设当前需要转移的节点为x,节点y,z分别是它的父亲,祖父,x需要转移到节点rt的下方。由于每一次Rotate操作每一次可以使节点上移一层(目的一般不会是下移),如果z就是rt,就说明y是x要到达的地方(因为z的下面就是y),而x到y只需要一次Rotate,因此调用单旋。
其他情况下至少需要两次Rotate操作,即双旋。直到到达目标位置为止。
如何判断是左旋还是右旋?
我们很容易发现一个规律——如果要使V上移到U(U是V的父亲),当V是U的左儿子时,我们需要右旋,而V是U的右儿子时,需要左旋……也就是说儿子的左右和旋转方向的左右是恰好相反的。
(3)查找树中是否存在某个节点
这是所有操作中最简单的一个,只用到了二叉排序树的性质。
设查找点的值为val,从根节点开始查找,设当前查找到的点值为u。由于根的左子树小于根,而右子树大于根,所以u>val时向左子树查询,否则向右子树查询,直到查找到值或者当前节点为空NULL。
(4)插入一个特定值的节点
基本思想和查找节点很像,也是根据二叉排序树来确定位置。
当我们找到一个值恰好为特定值的节点,则将该节点的个数+1,不再插入节点了。与查找不同的是它如果按顺序查找节点,发现该节点为NULL,就说明没有值为val的节点,此时我们会新建一个值为val的节点插入到那个为NULL的节点。
(5)查询点排名以及特定排名的点
这里的排名不包括并列(2,3,3,4 3的排名为2或3,4的排名为4)。其实就是比他小(严格小于)的元素个数+1,而比他小的元素恰好就是他的左子树,因此也就是它的左子树的个数+1。
查找特定排名的点要麻烦一些……设当前节点为u,当u的左子树+1大于排名,则说明当前数过大,向左子树查询,否则向右子树查询。若查询右子树,则先将特定排名减去当前节点的左子树大小,表示在右子树中需要找到第"特定排名减去当前节点的左子树大小"大的元素。
换句话说,当前节点为u,向u的右子树查询,则目标节点在u的右子树中的排名为 (以u为根的子树中的排名 - u的左子树大小)。
(6)删除特定值的点
还是先像查找特定值的节点的思路,先找到要删除的节点的位置。由于我把值相同的点压缩在了一个点上,值相同的点的个数为cnt。当cnt>1时,即不止一个点值为特定值,我们可以直接cnt--;如果cnt=1,则删除该点后,该点就没了……这时候我们需要处理节点与其前驱后继的关系。我们可以把前驱通过Splay移动到根节点,而把后继移到前驱的右儿子。我们会发现后继的左儿子就是要删除的节点,且它没有儿子(叶结点),所以我们直接把左儿子改为NULL,再Update更新节点个数,好像就完了(=^-ω-^=)
◇手打简单模板
(PS.下面这个模板实现了插入Insert,删除Delete(无法判断是否存在该元素),查找节点GetKey,正反向查询排名Find_Count/Get_Num,查找前驱后继FrontBehind)
1 /*Lucky_Glass*/ 2 #include<cstdio> 3 #include<cstring> 4 #include<algorithm> 5 using namespace std; 6 struct Node{ 7 Node *ch[2],*fa; //ch[0]左儿子,ch[1]右儿子 8 int v,cnt,size; //v点权,cnt点权为v的点数量,子树大小(包括根节点) 9 Node(int v):v(v){ //初始化 10 fa=ch[0]=ch[1]=NULL; 11 cnt=1; 12 } 13 int cmp(int x)const { //某时候极其方便的比较函数 14 if(x==v) return -1; 15 return x<v? 0:1; 16 } 17 }*root; //树根 18 int Get_Size(Node *p) //避免点为NULL时访问size错误 19 { 20 return p==NULL? 0:p->size; 21 } 22 void Update(Node *x) //上传子树大小 23 { 24 x->size=1+Get_Size(x->ch[0])+Get_Size(x->ch[1]); 25 } 26 void Rotate(Node *x,int d) //旋转,d=0左旋,d=1右旋 27 { 28 Node *y=x->fa; 29 y->ch[!d]=x->ch[d];x->fa=y->fa; 30 if(x->ch[d]!=NULL) x->ch[d]->fa=y; 31 if(y->fa!=NULL) y->fa->ch[y->fa->ch[1]==y]=x; 32 y->fa=x;x->ch[d]=y; 33 if(y==root) root=x; 34 Update(y); 35 } 36 void Splay(Node *x,Node *rt) 37 { 38 while(x->fa!=rt) //直到到达目标位置为止 39 { 40 Node *y=x->fa;Node *z=y->fa; 41 if(z==rt) //只旋转一次即到目标位置 42 if(x==y->ch[0]) Rotate(x,1); 43 else Rotate(x,0); 44 else //双旋 45 if(y==z->ch[0]) 46 if(x==y->ch[0]) 47 Rotate(y,1),Rotate(x,1); 48 else 49 Rotate(x,0),Rotate(x,1); 50 else 51 if(x==y->ch[1]) 52 Rotate(y,0),Rotate(x,0); 53 else 54 Rotate(x,1),Rotate(x,0); 55 } 56 Update(x); 57 } 58 void Insert(int val) //插入值为val的节点 59 { 60 if(root==NULL) {root=new Node(val);return;} 61 //插入节点 62 Node *y=root; 63 while(true) 64 { 65 if(val==y->v) {y->cnt++;Splay(y,NULL);return;} 66 //如果已经存在值为val的节点,则该节点个数+1 67 Node *&ch=(val<y->v? y->ch[0]:y->ch[1]); 68 if(ch==NULL) break; 69 y=ch; 70 } 71 Node *x=new Node(val); 72 (val<y->v? y->ch[0]:y->ch[1])=x; 73 x->fa=y; 74 Splay(x,NULL); 75 } 76 Node *Find(Node *x,int d) //寻找前驱后继(d=0前驱,d=1后继),只能寻找已存在于树中的值 77 { 78 while(x && x->ch[d]) x=x->ch[d]; 79 return x; 80 } 81 void Delete(int num) //删除一个值为num的节点 82 { 83 Node *p=root; 84 while(true) 85 { 86 if(!p) return; 87 if(p->v==num) 88 { 89 Splay(p,NULL); 90 if(p->cnt==1) //单个节点 91 { 92 Node *Front=Find(p->ch[0],1), 93 *Behind=Find(p->ch[1],0); //处理前驱后继 94 if(!Front && !Behind) root=NULL; 95 else if(!Front) root=root->ch[1],root->fa=NULL; 96 else if(!Behind) root=root->ch[0],root->fa=NULL; 97 else 98 { 99 Splay(Front,NULL); 100 Splay(Behind,root); 101 root->ch[1]->ch[0]=NULL; 102 root->ch[1]->size--; 103 } 104 } 105 else p->cnt--; //减少个数 106 return; 107 } 108 p=p->v>num? p->ch[0]:p->ch[1]; 109 } 110 } 111 Node *GetKey(Node *o,int x) //根据二叉排序树关系查找值为x的节点 112 { 113 int d=o->cmp(x); 114 if(d==-1) return o; 115 return GetKey(o->ch[d],x); 116 } 117 int Find_count(int val) //找到值为val的节点在树上的排名 118 { 119 Node *x=GetKey(root,val); 120 Splay(x,NULL); 121 return Get_Size(x->ch[0])+1; 122 } 123 int Get_Num(int num) //找到排名为num的数 124 { 125 Node *now=root; 126 while(now) 127 { 128 if(num>=Get_Size(now->ch[0])+1 && num<=Get_Size(now->ch[0])+now->cnt) 129 break; 130 if(Get_Size(now->ch[0])>=num) now=now->ch[0]; 131 else 132 { 133 num-=Get_Size(now->ch[0])+now->cnt; 134 now=now->ch[1]; 135 } 136 } 137 Splay(now,NULL); 138 return now->v; 139 } 140 int FrontBehind(int num,int d) //找前驱后继(不一定在树上) 141 { 142 Insert(num); 143 int res=Find(root->ch[d^1],d)->v; 144 Delete(num); 145 return res; 146 } 147 int main() 148 { 149 while(true) 150 { 151 int cmd,x; 152 scanf("%d%d",&cmd,&x); 153 switch(cmd) 154 { 155 case 1: Insert(x);break; 156 case 2: Delete(x);break; 157 case 3: printf("%d ",Find_count(x));break; 158 case 4: printf("%d ",Get_Num(x));break; 159 case 5: printf("%d ",FrontBehind(x,0));break; 160 case 6: printf("%d ",FrontBehind(x,1));break; 161 } 162 } 163 return 0; 164 }
这个代码风格可能比较奇怪,因为是从几个不同的代码裁剪修改,然后组合起来的……(∩╹□╹∩)
The End
Thanks for reading!
- Lucky_Glass