分类: C/C++
平衡二叉树——AVL树的实现
AVL树是最先发明的自平衡二叉查找算法,是平衡二叉树的一种。在AVL中任何节点的两个儿子子树的高度最大差别为1,所以它又被成为高度平衡树。查找、插入和删除在平均和最坏情况下都是O(log n)。增加和删除可能需要通过一次或多次树旋转来平衡这棵树。
假设把AVL树构造过程中需要重新平衡的节点叫做α。由于任意节点最多有两个儿子,因此高度不平衡时,α点的两颗子树的高度差2。这种不平衡可能出现在下面这四种情况:
1) 对α的左儿子的左子树进行一次插入(左旋)
其中D是新插入的节点,红色节点K2是失去平衡的节点。需要对K1和K2进行左旋调整即将K1作为根,将K2作为K1的左子树,K1的右子树调整为K2的左子树。如下图所示
进行左旋变换
代码如下:
- static Position SingleRotateWithLeft(Position K2)
- {
- Position K1;
- K1 = K2->Left;
- K2->Left = K1->Right;
- K1->Right = K2;
- //更新节点的高度
- return K1;
- }
2)对α的左儿子的右子树进行一次插入(左右双旋)
左右双旋这里的左右指的是对α的左儿子的右子树进行插入时需要旋转。先对K1和K2进行右旋(跟第四种情况类似),然后再对K3和K2进行左旋,最终实现平衡。如下图所示
进行一次右旋进行一次左旋
代码如下:
- static Position DoubleRotateWithLeft(Position K3)
- {
- K3->Left = SingleRotateWithRight(K3->Left);
- return SingleRotateWithLeft(K3);
- }
3)对α的右儿子的左子树进行一次插入(右左双旋)
右左双旋:先对K1和K2进行左旋,然后在对K2和K3进行右旋,最终实现平衡。如下图所示
进行一次左旋进行一次右旋
代码如下:
- static Position DoubleRotateWithRight(Position K3)
- {
- K3->Right = SingleRotateWithLeft(K3->Right);
- return SingleRotateWithRight(K3);
- }
4)对α的右儿子的右子树进行一次插入(右旋)
将K2的右子树更改为K1的左子树,K1的左子树更改为K2即完成的右旋,如下图所示
进行右旋
转载地址: http://blog.chinaunix.net/uid-24948645-id-3913917.html