AVL树是最先发明的自平衡二叉查找树,得名于它的发明者G.M. Adelson-Velsky和E.M. Landis。
在AVL树中任何节点的两个子树的高度最大差别为一,所以它也被称为高度平衡树。
查找、插入和删除在平均和最坏情况下都是O(log n),插入和删除可能需要通过一次或多次树旋转来重新平衡这个树。
如下图中,左测的树是AVL树,其操作效率明显高于右侧的树。
插入新节点可能会造成AVL树的失衡,应如何调整使其重新平衡呢?
首先要考虑所有不平衡的情形。
不平衡的情形:考虑3层,归为4类
新节点的插入位置有很多可能,怎么对不平衡的情形归类呢,或者说需要考虑多深的范围呢?
考虑三层高度即可。
下图中,假设对根(10)的左儿子(6)的左子树(4)进行一次插入,无论插入位置是左(插入2)还是右(插入5),都可以通过一次右旋转(顺时针旋转)完成平衡:
所以对根的左儿子的左子树进行一次插入后破坏平衡,都可归类为“左左”情况。
根据这样的思路,假设需要重新平衡的结点为α,可将所有可能归类为4种情况:
1.对α的左儿子的左子树进行一次插入(左左)
2.对α的左儿子的右子树进行一次插入(左右)
3.对α的右儿子的左子树进行一次插入(右左)
4.对α的右儿子的右子树进行一次插入(右右)
由于对称性,1、4和2、3是两组镜像问题,只要重点解决一侧的即可。下面考虑左左和左右的情况。
LL(左左)
速记方法:犯了两次左倾错误,需要一次右旋纠正。
在子树T0上添加新节点后,根节点g的左右子树失衡,g->T0 比 g->T3 深度多了2。
需对g节点执行一次右旋转,具体步骤为:
g.left = p.right; //g的左孩子指向p的右孩子
p.right = g; //g变成p的右孩子
root = p; //p作为新的根
LR(左右)
速记方法:犯了一次左倾错误和一次右倾错误,先左旋纠正右倾,再右旋纠正左倾。
在子树T2上添加新节点后,根节点g的左右子树失衡,g->T2 比 g->T3 深度多了2。
这时先对p节点执行一次左旋转(虽然以p为根的树平衡并未打破,但这是必要的准备工作),具体步骤为:
p.right = n.left; //p的左孩子指向n的右孩子
n.left = p; //p变成n的左孩子
root = n; //n作为子树新的根
经过一次左旋后,g的左孩子的左子树“过重”失衡了,可以归于“左左”的情形。(假设新节点插入位置是子树T1,左旋后仍是“左左”的情况)
再对g节点执行一次右旋转即可重新平衡:
g.left = p.right; //g的左孩子指向p的右孩子
p.right = g; //g变成p的右孩子
root = p; //p作为新的根
RR(右右)
左左的镜像问题。
只需对g执行一次左旋(逆时针旋转):
g.right = p.left; //g的右孩子指向p的左孩子
p.left = g; //g变成p的左孩子
root = p; //p作为新的根
RL(右左)
左右的镜像问题。
先对p节点执行一次右旋转:
p.left = n.right; //p的右孩子指向n的左孩子
n.right = p; //p变成n的右孩子
root = n; //n作为子树新的根
再对g节点执行一次左旋转即可重新平衡:
g.right = p.left; //g的右孩子指向p的左孩子
p.left = g; //g变成p的左孩子
root = p; //p作为新的根
代码实现
这里参考柳神的代码(本人写不出更漂亮的了…),输入一系列节点值,输出为根的值:
#include <iostream>
using namespace std;
struct node {
int val;
struct node *left, *right;
};
node *rotateLeft(node *root) {
node *t = root->right;
root->right = t->left;
t->left = root;
return t;
}
node *rotateRight(node *root) {
node *t = root->left;
root->left = t->right;
t->right = root;
return t;
}
node *rotateLeftRight(node *root) {
root->left = rotateLeft(root->left);
return rotateRight(root);
}
node *rotateRightLeft(node *root) {
root->right = rotateRight(root->right);
return rotateLeft(root);
}
int getHeight(node *root) {
if(root == NULL) return 0;
return max(getHeight(root->left), getHeight(root->right)) + 1;
}
node *insert(node *root, int val) {
if(root == NULL) {
root = new node();
root->val = val;
root->left = root->right = NULL;
} else if(val < root->val) {
root->left = insert(root->left, val);
if(getHeight(root->left) - getHeight(root->right) == 2)
root = val < root->left->val ? rotateRight(root) : rotateLeftRight(root);
} else {
root->right = insert(root->right, val);
if(getHeight(root->left) - getHeight(root->right) == -2)
root = val > root->right->val ? rotateLeft(root) : rotateRightLeft(root);
}
return root;
}
int main() {
int n, val;
scanf("%d", &n);
node *root = NULL;
for(int i = 0; i < n; i++) {
scanf("%d", &val);
root = insert(root, val);
}
printf("%d", root->val);
return 0;
}