目录
二叉搜索树虽可以缩短查找的效率,但如果数据有序或接近有序二叉搜索树将退化为单支树,查找元素相当于在顺序表中搜索元素,效率低下。
AVL树是解决上述问题的方法之一:
当向二叉搜索树插入新结点后,通过对树中的结点进行调整,保证每个结点的左右子树高度之差的绝对值不超过1,即可降低树的高度,从而减少平均搜索长度。
左右子树都是AVL树
左右子树高度之差(简称平衡因子)的绝对值不超过1
如果它有N个结点,其高度可保持在
,搜索时间复杂度O(
)
相比较于搜索二叉树,我们在实现 AVL 树时,用到了三叉连,每个结点除了左右指针,还有指向父亲结点的指针,还加入了平衡因子,方便判断和及时调整(旋转)
- template<class k,class v>
-
- //定义结点 ---三叉链,涉及到回溯问题
- struct AVLTreeNode
- {
- pair
_kv; - AVLTreeNode
* _left; - AVLTreeNode
* _right; - AVLTreeNode
* _parent; -
- int _bf; //平衡因子--右子树-左子树的高度差
-
- AVLTreeNode(const pair
& 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

左右双旋之后,可以看到三个结点的平衡因子都是 0

左右双旋之后,parent 和 subLR 平衡因子为 0 ,subL的平衡因子为 -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
于左右双旋道理相同,先右单旋,再左单旋,有三种情况:

如图示,右左双旋后,三个结点的平衡因子全部更新为 0

右左双旋之后,subR 和 subRL 平衡因子为 0 ,parent的平衡因子为 -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
& 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;
-
- }