转自: http://dongxicheng.org/structure/avl/
1. 概述
AVL树是最早提出的自平衡二叉树,在AVL树中任何节点的两个子树的高度最大差别为一,所以它也被称为高度平衡树。AVL树得名于它的发明者G.M. Adelson-Velsky和E.M. Landis。AVL树种查找、插入和删除在平均和最坏情况下都是O(log n),增加和删除可能需要通过一次或多次树旋转来重新平衡这个树。本文介绍了AVL树的设计思想和基本操作。
2. 基本术语
有四种种情况可能导致二叉查找树不平衡,分别为:
(1)LL:插入一个新节点到根节点的左子树(Left)的左子树(Left),导致根节点的平衡因子由1变为2
(2)RR:插入一个新节点到根节点的右子树(Right)的右子树(Right),导致根节点的平衡因子由-1变为-2
(3)LR:插入一个新节点到根节点的左子树(Left)的右子树(Right),导致根节点的平衡因子由1变为2
(4)RL:插入一个新节点到根节点的右子树(Right)的左子树(Left),导致根节点的平衡因子由-1变为-2
针对四种种情况可能导致的不平衡,可以通过旋转使之变平衡。有两种基本的旋转:
(1)左旋转:将根节点旋转到(根节点的)右孩子的左孩子位置
(2)右旋转:将根节点旋转到(根节点的)左孩子的右孩子位置
3. AVL树的旋转操作
AVL树的基本操作是旋转,有四种旋转方式,分别为:左旋转,右旋转,左右旋转(先左后右),右左旋转(先右后左),实际上,这四种旋转操作两两对称,因而也可以说成两类旋转操作。
基本的数据结构:
1
2
3
4
5
6
7
8
9
10
11
12
13
|
typedef
struct
Node* Tree;
typedef
struct
Node* Node_t;
typedef
Type
int
;
struct
Node{
Node_t left;
Node_t right;
int
height;
Type data;
};
int
Height(Node_t node) {
return
node->height;
}
|
3.1 LL
LL情况需要右旋解决,如下图所示:
代码为:
1
2
3
4
5
6
7
8
|
Node_t RightRotate(Node_t a) {
b = a->left;
a->left = b->right;
b->right = a;
a->height = Max(Height(a->left), Height(a->right));
b->height = Max(Height(b->left), Height(b->right));
return
b;
}
|
3.2 RR
RR情况需要左旋解决,如下图所示:
代码为:
1
2
3
4
5
6
7
8
|
Node_t LeftRotate(Node_t a) {
b = a->right;
a->right = b->left;
b->left = a;
a->height = Max(Height(a->left), Height(a->right));
b->height = Max(Height(b->left), Height(b->right));
return
b;
}
|
3.3 LR
LR情况需要左右(先B左旋转,后A右旋转)旋解决,如下图所示:
代码为:
1
2
3
4
|
Node_t LeftRightRotate(Node_t a) {
a->left = LeftRotate(a->left);
return
RightRotate(a);
}
|
3.4 RL
RL情况需要右左旋解决(先B右旋转,后A左旋转),如下图所示:
代码为:
1
2
3
4
|
Node_t RightLeftRotate(Node_t a) {
a->right = RightRotate(a->right);
return
LeftRotate(a);
}
|
4. AVL数的插入和删除操作
(1) 插入操作:实际上就是在不同情况下采用不同的旋转方式调整整棵树,具体代码如下:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
Node_t Insert(Type x, Tree t) {
if
(t == NULL) {
t = NewNode(x);
}
else
if
(x < t->data) {
t->left = Insert(t->left);
if
(Height(t->left) - Height(t->right) == 2) {
if
(x < t->left->data) {
t = RightRotate(t);
}
else
{
t = LeftRightRotate(t);
}
}
}
else
{
t->right = Insert(t->right);
if
(Height(t->right) - Height(t->left) == 2) {
if
(x > t->right->data) {
t = LeftRotate(t);
}
else
{
t = RightLeftRotate(t);
}
}
}
t->height = Max(Height(t->left), Height(t->right)) + 1;
return
t;
}
|
(2) 删除操作:首先定位要删除的节点,然后用该节点的右孩子的最左孩子替换该节点,并重新调整以该节点为根的子树为AVL树,具体调整方法跟插入数据类似,代码如下:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
Node_t Delete(Type x, Tree t) {
if
(t == NULL)
return
NULL;
if
(t->data == x) {
if
(t->right == NULL) {
Node_t temp = t;
t = t->left;
free
(temp);
}
else
{
Node_t head = t->right;
while
(head->left) {
head = head->left;
}
t->data = head->data;
//just copy data
t->right = Delete(t->data, t->right);
t->height = Max(Height(t->left), Height(t->right)) + 1;
}
return
t;
}
else
if
(t->data < x) {
Delete(x, t->right);
if
(t->right) Rotate(x, t->right);
}
else
{
Delete(x, t->left);
if
(t->left) Rotate(x, t->left);
}
if
(t) Rotate(x, t);
}
|
5. 总结
AVL树是最早的自平衡二叉树,相比于后来出现的平衡二叉树(红黑树,treap,splay树)而言,它现在应用较少,但研究AVL树对于了解后面出现的常用平衡二叉树具有重要意义。
6. 参考资料
(1) 数据结构(C语言版) 严蔚敏,吴伟民著
(2) http://zh.wikipedia.org/wiki/AVL%E6%A0%91
(3)http://www.cppblog.com/goodwin/archive/2011/08/08/152797.html
(4)http://www.asiteof.me/2010/06/avl/
———————————————————————————————-
更多关于数据结构和算法的介绍,请查看:数据结构与算法汇总
———————————————————————————————-
原创文章,转载请注明: 转载自董的博客
作者:Dong,作者介绍:http://dongxicheng.org/about/
本博客的文章集合:http://dongxicheng.org/recommend/