目录
二叉搜索树虽可以缩短查找的效率，但如果数据有序或接近有序二叉搜索树将退化为单支树，查找元素相当于在顺序表中搜索元素，效率低下。
AVL树
AVL树是解决上述问题的方法之一：
当向二叉搜索树插入新结点后，通过对树中的结点进行调整，保证每个结点的左右子树高度之差的绝对值不超过1，即可降低树的高度，从而减少平均搜索长度。
性质
左右子树都是AVL树
左右子树高度之差(简称平衡因子)的绝对值不超过1
如果它有N个结点，其高度可保持在 ，搜索时间复杂度O(
)
相比较于搜索二叉树，我们在实现 AVL 树时，用到了三叉连，每个结点除了左右指针，还有指向父亲结点的指针，还加入了平衡因子，方便判断和及时调整（旋转）
实现
结点构造
template<class k,class v>
//定义结点 ---三叉链，涉及到回溯问题
struct AVLTreeNode
{
pair<k, v> _kv;
AVLTreeNode<k, v>* _left;
AVLTreeNode<k, v>* _right;
AVLTreeNode<k, v>* _parent;
int _bf; //平衡因子--右子树-左子树的高度差
AVLTreeNode(const pair<k, v>& kv)
:_kv(kv)
, _left(nullptr)
, _right(nullptr)
, _parent(nullptr)
,_bf(0)
{}
};
在AVL树中，我们最需要搞明白的就是插入结点时，是怎么样保证左右高度差不超过 1 的？通过旋转来对树左右高度进行控制
左边高，向右旋转：右单旋
右边高，向左旋转：左单旋
左单旋
那么什么时候会进行左单旋呢？
当父亲结点平衡因子为 2 ，且其右子树结点平衡因子为 1
这颗树就不符合AVL树的规则，需要进行旋转，左单旋是右边高，向左进行旋转：
旋转很简单，但是每个结点都有三个指针，之间的关系很复杂，且还要考虑各种情况，上述是 subRL 为空的情况，且parent为根的情况。最后更新平衡因子，三个结点都为 0
代码实现、有注释说明：
//左单旋
void RotateL(Node* parent)
{
Node* subR = parent->_right;
Node* subRL = subR->_left;
parent->_right = subRL;
if (subRL)
subRL->_parent = parent; //当 subRL 不为空时，要注意结点父亲要更新
Node* ppNode = parent->_parent; //先保留父亲的父亲，以防父亲不是根节点
subR->_left = parent;
parent->_parent = subR; //注意父亲结点的父亲也要更新
if (parent == _root) //父亲为根，
{
_root = subR; //更新根
_root->_parent = nullptr; //根的父亲直接为空
}
else //如果父亲不为根，是上面其他结点的一个子树
{
if (parent == ppNode->_left) //判断是左子树
{
ppNode->_left = subR; //保留的结点起作用，保留的结点的指向更新
}
else
{
ppNode->_right = subR; 判断是右子树
}
subR->_parent = ppNode; //subR的父亲为保留的结点
}
subR->_bf = parent->_bf = 0; //最后平衡因子根据图更新
}右单旋
什么时候会进行右单旋呢？
当父亲结点平衡因子为 -2 ，且其左子树结点平衡因子为 -1
这颗树就不符合AVL树的规则，需要进行旋转，右单旋是左边高，向右进行旋转：
上述是 subLR 不为空，parent 为根的情况，旋转和平衡因子的更新如上图所示；
无论是左单旋还是右单旋，最后平衡因子的更新是发生在 parent 和 它的孩子结点上，都为 0
代码实现，有注释说明
//右单旋
void RotateR(Node* parent)
{
Node* subL = parent->_left;
Node* subLR = subL->_right;
parent->_left = subLR; //按图上规则，改变指向
if (subLR)
subLR->_parent = parent; //如果不为空，注意其父亲也应更新
Node* ppNode = parent->_parent; //保存父亲的父亲，防止父亲不是根节点，后面需要更新
subL->_right = parent; //如图，改变指向
parent->_parent = subL;
if (parent == _root) //如果父亲就是根节点
{
_root = subL;
_root->_parent = nullptr; //更新根节点，根节点指向空
}
else //不为根，那么就是上层结点的子树
{
if (parent == ppNode->_left) //判断为左子树
{
ppNode->_left = subL; //将上层结点的左指向我们的subL
}
else
{
ppNode->_right = subL; //判断为右子树，更新上层结点的右子树指向
}
subL->_parent = ppNode; //新根subL的父亲指向上层结点
}
subL->_bf = parent->_bf = 0; //最后更新原父亲和其孩子的平衡因子
}无论是左单旋还是右单旋，都是一边高的情况；
对于一个父亲节点来说，也会出现：
左子树高，左子树无左树有右树
右子树高，右子树无右树有左树
单旋并不能解决上述情况
这就是要处理的双旋转问题
单旋就处理不了吗？对的，处理不了，如下所示：
每种双旋转的大情况都有三种，涉及到平衡因子的更新，在此直接揭晓：
平衡因子的更新，取决于孩子的孩子 (孙字辈) 的平衡因子
左右双旋
什么时候会进行左右双旋呢？
当父亲结点平衡因子为 -2 ，且其左子树结点平衡因子为 1
1. 孙字辈的平衡因子为 0
左右双旋之后，可以看到三个结点的平衡因子都是 0
2. 孙字辈的平衡因子为 1
左右双旋之后，parent 和 subLR 平衡因子为 0 ，subL的平衡因子为 -1
3. 孙字辈的平衡因子为 -1
左右双旋之后，subL 和 subLR 平衡因子为 0 ，parent的平衡因子为 1
代码实现
//左右双旋
void RotateLR(Node* parent)
{
//记录
Node* subL = parent->_left;
Node* subLR = subL->_right;
int bf = subLR->_bf;
RotateL(parent->_left);
RotateR(parent); //平衡因子被全部调零，更新平衡因子
if (bf == 0) //根据孙字辈的结点平衡因子，对照图进行更新
{
parent->_bf = 0;
subL->_bf = 0;
subLR->_bf = 0;
}
else if (bf == 1)
{
parent->_bf = 0;
subL->_bf = -1;
subLR->_bf = 0;
}
else if (bf == -1)
{
parent->_bf = 1;
subL->_bf = 0;
subLR->_bf = 0;
}
else
{
//旋转前就有问题
assert(false);
}
}右左双旋
什么时候会进行右左双旋呢？
当父亲结点平衡因子为 2 ，且其右子树结点平衡因子为 -1
于左右双旋道理相同，先右单旋，再左单旋，有三种情况：
1. 孙字辈的平衡因子为 0
如图示，右左双旋后，三个结点的平衡因子全部更新为 0
2. 孙字辈的平衡因子为 1
右左双旋之后，subR 和 subRL 平衡因子为 0 ，parent的平衡因子为 -1
3. 孙字辈的平衡因子为 -1
右左双旋之后，parent 和 subRL 平衡因子为 0 ，subR的平衡因子为 1
代码实现
//右左双旋
void RotateRL(Node* parent)
{
//记录
Node* subR = parent->_right;
Node* subRL = subR->_left;
int bf = subRL->_bf; //记录孙字辈结点的平衡因子
RotateR(parent->_right); //根据上述分析，对相应结点进行旋转
RotateL(parent);
if (bf == 0) //根据孙字辈结点的平衡因子，进行更新
{
parent->_bf = 0;
subR->_bf = 0;
subRL->_bf = 0;
}
else if (bf == 1)
{
parent->_bf = -1;
subR->_bf = 0;
subRL->_bf = 0;
}
else if (bf == -1)
{
parent->_bf = 0;
subR->_bf = 1;
subRL->_bf = 0;
}
else
{
//旋转前就有问题
assert(false);
}
}插入的实现
AVL树的插入规则遵循搜索二叉树的插入规则，比结点大的插入在右边，比结点小的插入在左边，要插入，就应该先找到要插入的位置
代码实现--有相应细节注释说明
//插入
bool Insert(const pair<k,v>& kv)
{
//1. 按照搜索树的规则插入
//2. 看是否违反平衡规则，违反则旋转
//空树时
if (_root == nullptr)
{
_root = new Node(kv);
return true;
}
//找到插入位置
Node* parent = nullptr;
Node* cur = _root;
while (cur)
{
if (cur->_kv.first < kv.first)
{
parent = cur;
cur = cur->_right;
}
else if (cur->_kv.first > kv.first)
{
parent = cur;
cur = cur->_left;
}
else
{
return false;
}
}
cur = new Node(kv); //找到结点，创建结点，链接结点
if (parent->_kv.first < kv.first)
{
parent->_right = cur;
}
else
{
parent->_left = cur;
}
cur->_parent = parent; //注意结点的更新
// 已经将结点插入了，现在判断平衡因子--不影响另一颗子树，只会影响祖先
// （新增在父亲右，父亲平衡因子++，新增在左父亲平衡因子--）
//子树高度变了继续往上更新，高度不变停止更新
//没有违反规则，结束，违反规则--旋转处理
while (parent) //依次往上更新，最坏情况下最远更新到根
{
if (cur==parent->_right)
{
parent->_bf++;
}
else
{
parent->_bf--;
}
//是否继续更
if (parent->_bf == 0)
{
break; //高度不变
}
else if (parent->_bf == 1 || parent->_bf == -1)
{
//子树高度变了，继续网上更新祖先
cur = cur->_parent;
parent = parent->_parent;
}
else if (parent->_bf == 2 || parent->_bf == -2)
{
//子树不平衡旋转
if (parent->_bf == 2 && cur->_bf == 1) //左单旋
{
RotateL(parent);
}
else if (parent->_bf == -2 && cur->_bf == -1) //右单旋
{
RotateR(parent);
}
else if (parent->_bf == -2 && cur->_bf == 1) //左右双旋
{
RotateLR(parent);
}
else if (parent->_bf == 2 && cur->_bf == -1) //右左双旋
{
RotateRL(parent);
}
break;
}
else
{
//插入之前AVL就不平衡了
assert(false);
}
}
return true;
}
1221
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
