概要
本章介绍伸展树。它和"二叉查找树"和"AVL树"一样,都是特殊的二叉树。在了解了"二叉查找树"和"AVL树"之后,学习伸展树是一件相当容易的事情。和以往一样,本文会先对伸展树的理论知识进行简单介绍,然后给出C语言的实现。后序再分别给出C++和Java版本的实现;这3种实现方式的原理都一样,选择其中之一进行了解即可。若文章有错误或不足的地方,希望您能不吝指出!
目录
1. 伸展树的介绍
2. 伸展树的C实现
3. 伸展树的C测试程序
转载请注明出处:http://www.cnblogs.com/skywang12345/p/3604238.html
更多内容: 数据结构与算法系列 目录
(01) 伸展树(一)之 图文解析 和 C语言的实现
(02) 伸展树(二)之 C++的实现
(03) 伸展树(三)之 Java的实现
伸展树的介绍
伸展树(Splay Tree)是一种二叉排序树,它能在O(log n)内完成插入、查找和删除操作。它由Daniel Sleator和Robert Tarjan创造。
(01)
伸展树属于二叉查找树,即它具有和二叉查找树一样的性质:假设x为树中的任意一个结点,x节点包含关键字key,节点x的key值记为key[x]。如果y是x的左子树中的一个结点,则key[y]
<= key[x];如果y是x的右子树的一个结点,则key[y] >= key[x]。
(02) 除了拥有二叉查找树的性质之外,伸展树还具有的一个特点是:当某个节点被访问时,伸展树会通过旋转使该节点成为树根。这样做的好处是,下次要访问该节点时,能够迅速的访问到该节点。
假设想要对一个二叉查找树执行一系列的查找操作。为了使整个查找时间更小,被查频率高的那些条目就应当经常处于靠近树根的位置。于是想到设计一个简单方法,在每次查找之后对树进行重构,把被查找的条目搬移到离树根近一些的地方。伸展树应运而生,它是一种自调整形式的二叉查找树,它会沿着从某个节点到树根之间的路径,通过一系列的旋转把这个节点搬移到树根去。
相比于"二叉查找树"和"AVL树",学习伸展树时需要重点关注是"伸展树的旋转算法"。
伸展树的C实现
1. 节点定义
typedef int Type;
typedef struct SplayTreeNode {
Type key; // 关键字(键值)
struct SplayTreeNode *left; // 左孩子
struct SplayTreeNode *right; // 右孩子
} Node, *SplayTree;
伸展树的节点包括的几个组成元素:
(01) key -- 是关键字,是用来对伸展树的节点进行排序的。
(02) left -- 是左孩子。
(03) right -- 是右孩子。
外部接口
// 前序遍历"伸展树"
void preorder_splaytree(SplayTree tree);
// 中序遍历"伸展树"
void inorder_splaytree(SplayTree tree);
// 后序遍历"伸展树"
void postorder_splaytree(SplayTree tree);
// (递归实现)查找"伸展树x"中键值为key的节点
Node* splaytree_search(SplayTree x, Type key);
// (非递归实现)查找"伸展树x"中键值为key的节点
Node* iterative_splaytree_search(SplayTree x, Type key);
// 查找最小结点:返回tree为根结点的伸展树的最小结点。
Node* splaytree_minimum(SplayTree tree);
// 查找最大结点:返回tree为根结点的伸展树的最大结点。
Node* splaytree_maximum(SplayTree tree);
// 旋转key对应的节点为根节点。
Node* splaytree_splay(SplayTree tree, Type key);
// 将结点插入到伸展树中,并返回根节点
Node* insert_splaytree(SplayTree tree, Type key);
// 删除结点(key为节点的值),并返回根节点
Node* delete_splaytree(SplayTree tree, Type key);
// 销毁伸展树
void destroy_splaytree(SplayTree tree);
// 打印伸展树
void print_splaytree(SplayTree tree, Type key, int direction);
2. 旋转
旋转的代码
/*
* 旋转key对应的节点为根节点,并返回根节点。
*
* 注意:
* (a):伸展树中存在"键值为key的节点"。
* 将"键值为key的节点"旋转为根节点。
* (b):伸展树中不存在"键值为key的节点",并且key < tree->key。
* b-1 "键值为key的节点"的前驱节点存在的话,将"键值为key的节点"的前驱节点旋转为根节点。
* b-2 "键值为key的节点"的前驱节点存在的话,则意味着,key比树中任何键值都小,那么此时,将最小节点旋转为根节点。
* (c):伸展树中不存在"键值为key的节点",并且key > tree->key。
* c-1 "键值为key的节点"的后继节点存在的话,将"键值为key的节点"的后继节点旋转为根节点。
* c-2 "键值为key的节点"的后继节点不存在的话,则意味着,key比树中任何键值都大,那么此时,将最大节点旋转为根节点。
*/
Node* splaytree_splay(SplayTree tree, Type key)
{
Node N, *l, *r, *c;
if (tree == NULL)
return tree;
N.left = N.right = NULL;
l = r = &N;
for (;;)
{
if (key < tree->key)
{
if (tree->left == NULL)
break;
if (key < tree->left->key)
{
c = tree->left; /* 01, rotate right */
tree->left = c->right;
c->right = tree;
tree = c;
if (tree->left == NULL)
break;
}
r->left = tree; /* 02, link right */
r = tree;
tree = tree->left;
}
else if (key > tree->key)
{
if (tree->right == NULL)
break;
if (key > tree->right->key)
{
c = tree->right; /* 03, rotate left */
tree->right = c->left;
c->left = tree;
tree = c;
if (tree->right == NULL)
break;
}
l->right = tree; /* 04, link left */
l = tree;
tree = tree->right;
}
else
{
break;
}
}
l->right = tree->left; /* 05, assemble */
r->left = tree->right;
tree->left = N.right;
tree->right = N.left;
return tree;
}
上面的代码的作用:将"键值为key的节点"旋转为根节点,并返回根节点。它的处理情况共包括:
(a):伸展树中存在"键值为key的节点"。
将"键值为key的节点"旋转为根节点。
(b):伸展树中不存在"键值为key的节点",并且key < tree->key。
b-1) "键值为key的节点"的前驱节点存在的话,将"键值为key的节点"的前驱节点旋转为根节点。
b-2) "键值为key的节点"的前驱节点存在的话,则意味着,key比树中任何键值都小,那么此时,将最小节点旋转为根节点。
(c):伸展树中不存在"键值为key的节点",并且key > tree->key。
c-1) "键值为key的节点"的后继节点存在的话,将"键值为key的节点"的后继节点旋转为根节点。
c-2) "键值为key的节点"的后继节点不存在的话,则意味着,key比树中任何键值都大,那么此时,将最大节点旋转为根节点。
下面列举个例子分别对a进行说明。
在下面的伸展树中查找10,共包括"右旋" --> "右链接" --> "组合"这3步。
第一步: 右旋
对应代码中的"rotate right"部分
第二步: 右链接
对应代码中的"link right"部分
第三步: 组合
对应代码中的"assemble"部分
提示:如果在上面的伸展树中查找"70",则正好与"示例1"对称,而对应的操作则分别是"rotate left", "link left"和"assemble"。
其它的情况,例如"查找15是b-1的情况,查找5是b-2的情况"等等,这些都比较简单,大家可以自己分析。
3. 插入
/*
* 将结点插入到伸展树中(不旋转)
*
* 参数说明:
* tree 伸展树的根结点
* z 插入的结点
* 返回值:
* 根节点
*/
static Node* splaytree_insert(SplayTree tree, Node *z)
{
Node *y = NULL;
Node *x = tree;
// 查找z的插入位置
while (x != NULL)
{
y = x;
if (z->key < x->key)
x = x->left;
else if (z->key > x->key)
x = x->right;
else
{
printf("不允许插入相同节点(%d)!\n", z->key);
// 释放申请的节点,并返回tree。
free(z);
return tree;
}
}
if (y==NULL)
tree = z;
else if (z->key < y->key)
y->left = z;
else
y->right = z;
return tree;
}
/*
* 创建并返回伸展树结点。
*
* 参数说明:
* key 是键值。
* parent 是父结点。
* left 是左孩子。
* right 是右孩子。
*/
static Node* create_splaytree_node(Type key, Node *left, Node* right)
{
Node* p;
if ((p = (Node *)malloc(sizeof(Node))) == NULL)
return NULL;
p->key = key;
p->left = left;
p->right = right;
return p;
}
/*
* 新建结点(key),然后将其插入到伸展树中,并将插入节点旋转为根节点
*
* 参数说明:
* tree 伸展树的根结点
* key 插入结点的键值
* 返回值:
* 根节点
*/
Node* insert_splaytree(SplayTree tree, Type key)
{
Node *z; // 新建结点
// 如果新建结点失败,则返回。
if ((z=create_splaytree_node(key, NULL, NULL)) == NULL)
return tree;
// 插入节点
tree = splaytree_insert(tree, z);
// 将节点(key)旋转为根节点
tree = splaytree_splay(tree, key);
}
外部接口: insert_splaytree(tree, key)是提供给外部的接口,它的作用是新建节点(节点的键值为key),并将节点插入到伸展树中;然后,将该节点旋转为根节点。
内部接口: splaytree_insert(tree, z)是内部接口,它的作用是将节点z插入到tree中。splaytree_insert(tree, z)在将z插入到tree中时,仅仅只将tree当作是一棵二叉查找树,而且不允许插入相同节点。
4. 删除
删除接口
/*
* 删除结点(key为节点的键值),并返回根节点。
*
* 参数说明:
* tree 伸展树的根结点
* z 删除的结点
* 返回值:
* 根节点(根节点是被删除节点的前驱节点)
*
*/
Node* delete_splaytree(SplayTree tree, Type key)
{
Node *x;
if (tree == NULL)
return NULL;
// 查找键值为key的节点,找不到的话直接返回。
if (splaytree_search(tree, key) == NULL)
return tree;
// 将key对应的节点旋转为根节点。
tree = splaytree_splay(tree, key);
if (tree->left != NULL)
{
// 将"tree的前驱节点"旋转为根节点
x = splaytree_splay(tree->left, key);
// 移除tree节点
x->right = tree->right;
}
else
x = tree->right;
free(tree);
return x;
}
delete_splaytree(tree, key)的作用是:删除伸展树中键值为key的节点。
它会先在伸展树中查找键值为key的节点。若没有找到的话,则直接返回。若找到的话,则将该节点旋转为根节点,然后再删除该节点。
注意:关于伸展树的"前序遍历"、"中序遍历"、"后序遍历"、"最大值"、"最小值"、"查找"、"打印"、"销毁"等接口与"二叉查找树"基本一样,这些操作在"二叉查找树"中已经介绍过了,这里就不再单独介绍了。当然,后文给出的伸展树的完整源码中,有给出这些API的实现代码。这些接口很简单,Please RTFSC(Read The Fucking Source Code)!
伸展树的C实现(完整源码)
伸展树的头文件(splay_tree.h)
1 #ifndef _SPLAY_TREE_H_
2 #define _SPLAY_TREE_H_
3
4 typedef int Type;
5
6 typedef struct SplayTreeNode {
7 Type key; // 关键字(键值)
8 struct SplayTreeNode *left; // 左孩子
9 struct SplayTreeNode *right; // 右孩子
10 } Node, *SplayTree;
11
12 // 前序遍历"伸展树"
13 void preorder_splaytree(SplayTree tree);
14 // 中序遍历"伸展树"
15 void inorder_splaytree(SplayTree tree);
16 // 后序遍历"伸展树"
17 void postorder_splaytree(SplayTree tree);
18
19 // (递归实现)查找"伸展树x"中键值为key的节点
20 Node* splaytree_search(SplayTree x, Type key);
21 // (非递归实现)查找"伸展树x"中键值为key的节点
22 Node* iterative_splaytree_search(SplayTree x, Type key);
23
24 // 查找最小结点:返回tree为根结点的伸展树的最小结点。
25 Node* splaytree_minimum(SplayTree tree);
26 // 查找最大结点:返回tree为根结点的伸展树的最大结点。
27 Node* splaytree_maximum(SplayTree tree);
28
29 // 旋转key对应的节点为根节点。
30 Node* splaytree_splay(SplayTree tree, Type key);
31
32 // 将结点插入到伸展树中,并返回根节点
33 Node* insert_splaytree(SplayTree tree, Type key);
34
35 // 删除结点(key为节点的值),并返回根节点
36 Node* delete_splaytree(SplayTree tree, Type key);
37
38 // 销毁伸展树
39 void destroy_splaytree(SplayTree tree);
40
41 // 打印伸展树
42 void print_splaytree(SplayTree tree, Type key, int direction);
43
44 #endif
伸展树的实现文件(splay_tree.c)
1 /**
2 * SplayTree伸展树(C语言): C语言实现的伸展树。
3 *
4 * @author skywang
5 * @date 2014/02/03
6 */
7
8 #include <stdio.h>
9 #include <stdlib.h>
10 #include "splay_tree.h"
11
12 /*
13 * 前序遍历"伸展树"
14 */
15 void preorder_splaytree(SplayTree tree)
16 {
17 if(tree != NULL)
18 {
19 printf("%d ", tree->key);
20 preorder_splaytree(tree->left);
21 preorder_splaytree(tree->right);
22 }
23 }
24
25 /*
26 * 中序遍历"伸展树"
27 */
28 void inorder_splaytree(SplayTree tree)
29 {
30 if(tree != NULL)
31 {
32 inorder_splaytree(tree->left);
33 printf("%d ", tree->key);
34 inorder_splaytree(tree->right);
35 }
36 }
37
38 /*
39 * 后序遍历"伸展树"
40 */
41 void postorder_splaytree(SplayTree tree)
42 {
43 if(tree != NULL)
44 {
45 postorder_splaytree(tree->left);
46 postorder_splaytree(tree->right);
47 printf("%d ", tree->key);
48 }
49 }
50
51 /*
52 * (递归实现)查找"伸展树x"中键值为key的节点
53 */
54 Node* splaytree_search(SplayTree x, Type key)
55 {
56 if (x==NULL || x->key==key)
57 return x;
58
59 if (key < x->key)
60 return splaytree_search(x->left, key);
61 else
62 return splaytree_search(x->right, key);
63 }
64
65 /*
66 * (非递归实现)查找"伸展树x"中键值为key的节点
67 */
68 Node* iterative_splaytree_search(SplayTree x, Type key)
69 {
70 while ((x!=NULL) && (x->key!=key))
71 {
72 if (key < x->key)
73 x = x->left;
74 else
75 x = x->right;
76 }
77
78 return x;
79 }
80
81 /*
82 * 查找最小结点:返回tree为根结点的伸展树的最小结点。
83 */
84 Node* splaytree_minimum(SplayTree tree)
85 {
86 if (tree == NULL)
87 return NULL;
88
89 while(tree->left != NULL)
90 tree = tree->left;
91 return tree;
92 }
93
94 /*
95 * 查找最大结点:返回tree为根结点的伸展树的最大结点。
96 */
97 Node* splaytree_maximum(SplayTree tree)
98 {
99 if (tree == NULL)
100 return NULL;
101
102 while(tree->right != NULL)
103 tree = tree->right;
104 return tree;
105 }
106
107 /*
108 * 旋转key对应的节点为根节点,并返回根节点。
109 *
110 * 注意:
111 * (a):伸展树中存在"键值为key的节点"。
112 * 将"键值为key的节点"旋转为根节点。
113 * (b):伸展树中不存在"键值为key的节点",并且key < tree->key。
114 * b-1 "键值为key的节点"的前驱节点存在的话,将"键值为key的节点"的前驱节点旋转为根节点。
115 * b-2 "键值为key的节点"的前驱节点存在的话,则意味着,key比树中任何键值都小,那么此时,将最小节点旋转为根节点。
116 * (c):伸展树中不存在"键值为key的节点",并且key > tree->key。
117 * c-1 "键值为key的节点"的后继节点存在的话,将"键值为key的节点"的后继节点旋转为根节点。
118 * c-2 "键值为key的节点"的后继节点不存在的话,则意味着,key比树中任何键值都大,那么此时,将最大节点旋转为根节点。
119 */
120 Node* splaytree_splay(SplayTree tree, Type key)
121 {
122 Node N, *l, *r, *c;
123
124 if (tree == NULL)
125 return tree;
126
127 N.left = N.right = NULL;
128 l = r = &N;
129
130 for (;;)
131 {
132 if (key < tree->key)
133 {
134 if (tree->left == NULL)
135 break;
136 if (key < tree->left->key)
137 {
138 c = tree->left; /* 01, rotate right */
139 tree->left = c->right;
140 c->right = tree;
141 tree = c;
142 if (tree->left == NULL)
143 break;
144 }
145 r->left = tree; /* 02, link right */
146 r = tree;
147 tree = tree->left;
148 }
149 else if (key > tree->key)
150 {
151 if (tree->right == NULL)
152 break;
153 if (key > tree->right->key)
154 {
155 c = tree->right; /* 03, rotate left */
156 tree->right = c->left;
157 c->left = tree;
158 tree = c;
159 if (tree->right == NULL)
160 break;
161 }
162 l->right = tree; /* 04, link left */
163 l = tree;
164 tree = tree->right;
165 }
166 else
167 {
168 break;
169 }
170 }
171
172 l->right = tree->left; /* 05, assemble */
173 r->left = tree->right;
174 tree->left = N.right;
175 tree->right = N.left;
176
177 return tree;
178 }
179
180 /*
181 * 将结点插入到伸展树中(不旋转)
182 *
183 * 参数说明:
184 * tree 伸展树的根结点
185 * z 插入的结点
186 * 返回值:
187 * 根节点
188 */
189 static Node* splaytree_insert(SplayTree tree, Node *z)
190 {
191 Node *y = NULL;
192 Node *x = tree;
193
194 // 查找z的插入位置
195 while (x != NULL)
196 {
197 y = x;
198 if (z->key < x->key)
199 x = x->left;
200 else if (z->key > x->key)
201 x = x->right;
202 else
203 {
204 printf("不允许插入相同节点(%d)!\n", z->key);
205 // 释放申请的节点,并返回tree。
206 free(z);
207 return tree;
208 }
209 }
210
211 if (y==NULL)
212 tree = z;
213 else if (z->key < y->key)
214 y->left = z;
215 else
216 y->right = z;
217
218 return tree;
219 }
220
221 /*
222 * 创建并返回伸展树结点。
223 *
224 * 参数说明:
225 * key 是键值。
226 * parent 是父结点。
227 * left 是左孩子。
228 * right 是右孩子。
229 */
230 static Node* create_splaytree_node(Type key, Node *left, Node* right)
231 {
232 Node* p;
233
234 if ((p = (Node *)malloc(sizeof(Node))) == NULL)
235 return NULL;
236 p->key = key;
237 p->left = left;
238 p->right = right;
239
240 return p;
241 }
242
243 /*
244 * 新建结点(key),然后将其插入到伸展树中,并将插入节点旋转为根节点
245 *
246 * 参数说明:
247 * tree 伸展树的根结点
248 * key 插入结点的键值
249 * 返回值:
250 * 根节点
251 */
252 Node* insert_splaytree(SplayTree tree, Type key)
253 {
254 Node *z; // 新建结点
255
256 // 如果新建结点失败,则返回。
257 if ((z=create_splaytree_node(key, NULL, NULL)) == NULL)
258 return tree;
259
260 // 插入节点
261 tree = splaytree_insert(tree, z);
262 // 将节点(key)旋转为根节点
263 tree = splaytree_splay(tree, key);
264 }
265
266 /*
267 * 删除结点(key为节点的键值),并返回根节点。
268 *
269 * 参数说明:
270 * tree 伸展树的根结点
271 * z 删除的结点
272 * 返回值:
273 * 根节点(根节点是被删除节点的前驱节点)
274 *
275 */
276 Node* delete_splaytree(SplayTree tree, Type key)
277 {
278 Node *x;
279
280 if (tree == NULL)
281 return NULL;
282
283 // 查找键值为key的节点,找不到的话直接返回。
284 if (splaytree_search(tree, key) == NULL)
285 return tree;
286
287 // 将key对应的节点旋转为根节点。
288 tree = splaytree_splay(tree, key);
289
290 if (tree->left != NULL)
291 {
292 // 将"tree的前驱节点"旋转为根节点
293 x = splaytree_splay(tree->left, key);
294 // 移除tree节点
295 x->right = tree->right;
296 }
297 else
298 x = tree->right;
299
300 free(tree);
301
302 return x;
303 }
304
305 /*
306 * 销毁伸展树
307 */
308 void destroy_splaytree(SplayTree tree)
309 {
310 if (tree==NULL)
311 return ;
312
313 if (tree->left != NULL)
314 destroy_splaytree(tree->left);
315 if (tree->right != NULL)
316 destroy_splaytree(tree->right);
317
318 free(tree);
319 }
320
321 /*
322 * 打印"伸展树"
323 *
324 * tree -- 伸展树的节点
325 * key -- 节点的键值
326 * direction -- 0,表示该节点是根节点;
327 * -1,表示该节点是它的父结点的左孩子;
328 * 1,表示该节点是它的父结点的右孩子。
329 */
330 void print_splaytree(SplayTree tree, Type key, int direction)
331 {
332 if(tree != NULL)
333 {
334 if(direction==0) // tree是根节点
335 printf("%2d is root\n", tree->key);
336 else // tree是分支节点
337 printf("%2d is %2d‘s %6s child\n", tree->key, key, direction==1?"right" : "left");
338
339 print_splaytree(tree->left, tree->key, -1);
340 print_splaytree(tree->right,tree->key, 1);
341 }
342 }
伸展树的测试程序(splaytree_test.c)
1 /**
2 * C 语言: 伸展树测试程序
3 *
4 * @author skywang
5 * @date 2014/02/03
6 */
7
8 #include <stdio.h>
9 #include "splay_tree.h"
10
11 static int arr[]= {10,50,40,30,20,60};
12 #define TBL_SIZE(a) ( (sizeof(a)) / (sizeof(a[0])) )
13
14 void main()
15 {
16 int i, ilen;
17 SplayTree root=NULL;
18
19 printf("== 依次添加: ");
20 ilen = TBL_SIZE(arr);
21 for(i=0; i<ilen; i++)
22 {
23 printf("%d ", arr[i]);
24 root = insert_splaytree(root, arr[i]);
25 }
26
27 printf("\n== 前序遍历: ");
28 preorder_splaytree(root);
29
30 printf("\n== 中序遍历: ");
31 inorder_splaytree(root);
32
33 printf("\n== 后序遍历: ");
34 postorder_splaytree(root);
35 printf("\n");
36
37 printf("== 最小值: %d\n", splaytree_minimum(root)->key);
38 printf("== 最大值: %d\n", splaytree_maximum(root)->key);
39 printf("== 树的详细信息: \n");
40 print_splaytree(root, root->key, 0);
41
42 i = 30;
43 printf("\n== 旋转节点(%d)为根节点\n", i);
44 printf("== 树的详细信息: \n");
45 root = splaytree_splay(root, i);
46 print_splaytree(root, root->key, 0);
47
48 // 销毁伸展树
49 destroy_splaytree(root);
50 }
伸展树的C测试程序
伸展树的测试程序运行结果如下:
== 依次添加: 10 50 40 30 20 60
== 前序遍历: 60 30 20 10 50 40
== 中序遍历: 10 20 30 40 50 60
== 后序遍历: 10 20 40 50 30 60
== 最小值: 10
== 最大值: 60
== 树的详细信息:
60 is root
30 is 60‘s left child
20 is 30‘s left child
10 is 20‘s left child
50 is 30‘s right child
40 is 50‘s left child
== 旋转节点(30)为根节点
== 树的详细信息:
30 is root
20 is 30‘s left child
10 is 20‘s left child
60 is 30‘s right child
50 is 60‘s left child
40 is 50‘s left child
测试程序的主要流程是:新建伸展树,然后向伸展树中依次插入10,50,40,30,20,60。插入完毕这些数据之后,伸展树的节点是60;此时,再旋转节点,使得30成为根节点。
依次插入10,50,40,30,20,60示意图如下:
将30旋转为根节点的示意图如下: