• AVL树的插入--旋转笔记


    目录

    AVL树

    性质

    实现

    结点构造

    左单旋

    代码实现、有注释说明:

    右单旋 

    代码实现,有注释说明

    左右双旋

    1. 孙字辈的平衡因子为 0

    2. 孙字辈的平衡因子为 1

    3.  孙字辈的平衡因子为 -1

    代码实现

    右左双旋

    1. 孙字辈的平衡因子为 0

    2. 孙字辈的平衡因子为 1

    3.  孙字辈的平衡因子为 -1

    代码实现

    插入的实现

    代码实现--有相应细节注释说明

     


    二叉搜索树虽可以缩短查找的效率,但如果数据有序或接近有序二叉搜索树将退化为单支树,查找元素相当于在顺序表中搜索元素,效率低下。

    AVL树

    AVL树是解决上述问题的方法之一:

    当向二叉搜索树插入新结点后,通过对树中的结点进行调整,保证每个结点的左右子树高度之差的绝对值不超过1,即可降低树的高度,从而减少平均搜索长度。

    性质

    左右子树都是AVL树
    左右子树高度之差(简称平衡因子)的绝对值不超过1
    如果它有N个结点,其高度可保持在 log2N,搜索时间复杂度O(log2N)

    相比较于搜索二叉树,我们在实现 AVL 树时,用到了三叉连,每个结点除了左右指针,还有指向父亲结点的指针,还加入了平衡因子,方便判断和及时调整(旋转)

    实现

    结点构造

    1. template<class k,class v>
    2. //定义结点 ---三叉链,涉及到回溯问题
    3. struct AVLTreeNode
    4. {
    5. pair _kv;
    6. AVLTreeNode* _left;
    7. AVLTreeNode* _right;
    8. AVLTreeNode* _parent;
    9. int _bf; //平衡因子--右子树-左子树的高度差
    10. AVLTreeNode(const pair& kv)
    11. :_kv(kv)
    12. , _left(nullptr)
    13. , _right(nullptr)
    14. , _parent(nullptr)
    15. ,_bf(0)
    16. {}
    17. };

    在AVL树中,我们最需要搞明白的就是插入结点时,是怎么样保证左右高度差不超过 1 的?通过旋转来对树左右高度进行控制

    左边高,向右旋转:右单旋

    右边高,向左旋转:左单旋

    左单旋

    那么什么时候会进行左单旋呢?

    当父亲结点平衡因子为 2 ,且其右子树结点平衡因子为 1

    这颗树就不符合AVL树的规则,需要进行旋转,左单旋是右边高,向左进行旋转:

    旋转很简单,但是每个结点都有三个指针,之间的关系很复杂,且还要考虑各种情况,上述是 subRL 为空的情况,且parent为根的情况。最后更新平衡因子,三个结点都为 0

    代码实现、有注释说明:

    1. //左单旋
    2. void RotateL(Node* parent)
    3. {
    4. Node* subR = parent->_right;
    5. Node* subRL = subR->_left;
    6. parent->_right = subRL;
    7. if (subRL)
    8. subRL->_parent = parent; //当 subRL 不为空时,要注意结点父亲要更新
    9. Node* ppNode = parent->_parent; //先保留父亲的父亲,以防父亲不是根节点
    10. subR->_left = parent;
    11. parent->_parent = subR; //注意父亲结点的父亲也要更新
    12. if (parent == _root) //父亲为根,
    13. {
    14. _root = subR; //更新根
    15. _root->_parent = nullptr; //根的父亲直接为空
    16. }
    17. else //如果父亲不为根,是上面其他结点的一个子树
    18. {
    19. if (parent == ppNode->_left) //判断是左子树
    20. {
    21. ppNode->_left = subR; //保留的结点起作用,保留的结点的指向更新
    22. }
    23. else
    24. {
    25. ppNode->_right = subR; 判断是右子树
    26. }
    27. subR->_parent = ppNode; //subR的父亲为保留的结点
    28. }
    29. subR->_bf = parent->_bf = 0; //最后平衡因子根据图更新
    30. }

    右单旋 

    什么时候会进行右单旋呢?

    当父亲结点平衡因子为 -2 ,且其左子树结点平衡因子为 -1

    这颗树就不符合AVL树的规则,需要进行旋转,右单旋是左边高,向右进行旋转:

    上述是 subLR 不为空,parent 为根的情况,旋转和平衡因子的更新如上图所示;

    无论是左单旋还是右单旋,最后平衡因子的更新是发生在 parent 和 它的孩子结点上,都为 0 

    代码实现,有注释说明

    1. //右单旋
    2. void RotateR(Node* parent)
    3. {
    4. Node* subL = parent->_left;
    5. Node* subLR = subL->_right;
    6. parent->_left = subLR; //按图上规则,改变指向
    7. if (subLR)
    8. subLR->_parent = parent; //如果不为空,注意其父亲也应更新
    9. Node* ppNode = parent->_parent; //保存父亲的父亲,防止父亲不是根节点,后面需要更新
    10. subL->_right = parent; //如图,改变指向
    11. parent->_parent = subL;
    12. if (parent == _root) //如果父亲就是根节点
    13. {
    14. _root = subL;
    15. _root->_parent = nullptr; //更新根节点,根节点指向空
    16. }
    17. else //不为根,那么就是上层结点的子树
    18. {
    19. if (parent == ppNode->_left) //判断为左子树
    20. {
    21. ppNode->_left = subL; //将上层结点的左指向我们的subL
    22. }
    23. else
    24. {
    25. ppNode->_right = subL; //判断为右子树,更新上层结点的右子树指向
    26. }
    27. subL->_parent = ppNode; //新根subL的父亲指向上层结点
    28. }
    29. subL->_bf = parent->_bf = 0; //最后更新原父亲和其孩子的平衡因子
    30. }

    无论是左单旋还是右单旋,都是一边高的情况;

    对于一个父亲节点来说,也会出现:

    左子树高,左子树无左树有右树

    右子树高,右子树无右树有左树

    单旋并不能解决上述情况

    这就是要处理的双旋转问题

    单旋就处理不了吗?对的,处理不了,如下所示:

    每种双旋转的大情况都有三种,涉及到平衡因子的更新,在此直接揭晓:

    平衡因子的更新,取决于孩子的孩子 (孙字辈) 的平衡因子

    左右双旋

    什么时候会进行左右双旋呢?

    当父亲结点平衡因子为 -2 ,且其左子树结点平衡因子为 1

    1. 孙字辈的平衡因子为 0

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

    2. 孙字辈的平衡因子为 1

    左右双旋之后,parent 和 subLR 平衡因子为 0 ,subL的平衡因子为  -1

    3.  孙字辈的平衡因子为 -1

    左右双旋之后,subL 和 subLR 平衡因子为 0 ,parent的平衡因子为  1

    代码实现

    1. //左右双旋
    2. void RotateLR(Node* parent)
    3. {
    4. //记录
    5. Node* subL = parent->_left;
    6. Node* subLR = subL->_right;
    7. int bf = subLR->_bf;
    8. RotateL(parent->_left);
    9. RotateR(parent); //平衡因子被全部调零,更新平衡因子
    10. if (bf == 0) //根据孙字辈的结点平衡因子,对照图进行更新
    11. {
    12. parent->_bf = 0;
    13. subL->_bf = 0;
    14. subLR->_bf = 0;
    15. }
    16. else if (bf == 1)
    17. {
    18. parent->_bf = 0;
    19. subL->_bf = -1;
    20. subLR->_bf = 0;
    21. }
    22. else if (bf == -1)
    23. {
    24. parent->_bf = 1;
    25. subL->_bf = 0;
    26. subLR->_bf = 0;
    27. }
    28. else
    29. {
    30. //旋转前就有问题
    31. assert(false);
    32. }
    33. }

    右左双旋

    什么时候会进行右左双旋呢?

    当父亲结点平衡因子为 2 ,且其右子树结点平衡因子为 -1

    于左右双旋道理相同,先右单旋,再左单旋,有三种情况:

    1. 孙字辈的平衡因子为 0

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

    2. 孙字辈的平衡因子为 1

    右左双旋之后,subR 和 subRL 平衡因子为 0 ,parent的平衡因子为  -1

    3.  孙字辈的平衡因子为 -1

    右左双旋之后,parent 和 subRL 平衡因子为 0 ,subR的平衡因子为  1

    代码实现

    1. //右左双旋
    2. void RotateRL(Node* parent)
    3. {
    4. //记录
    5. Node* subR = parent->_right;
    6. Node* subRL = subR->_left;
    7. int bf = subRL->_bf; //记录孙字辈结点的平衡因子
    8. RotateR(parent->_right); //根据上述分析,对相应结点进行旋转
    9. RotateL(parent);
    10. if (bf == 0) //根据孙字辈结点的平衡因子,进行更新
    11. {
    12. parent->_bf = 0;
    13. subR->_bf = 0;
    14. subRL->_bf = 0;
    15. }
    16. else if (bf == 1)
    17. {
    18. parent->_bf = -1;
    19. subR->_bf = 0;
    20. subRL->_bf = 0;
    21. }
    22. else if (bf == -1)
    23. {
    24. parent->_bf = 0;
    25. subR->_bf = 1;
    26. subRL->_bf = 0;
    27. }
    28. else
    29. {
    30. //旋转前就有问题
    31. assert(false);
    32. }
    33. }

    插入的实现

    AVL树的插入规则遵循搜索二叉树的插入规则,比结点大的插入在右边,比结点小的插入在左边,要插入,就应该先找到要插入的位置 

    代码实现--有相应细节注释说明

    1. //插入
    2. bool Insert(const pair& kv)
    3. {
    4. //1. 按照搜索树的规则插入
    5. //2. 看是否违反平衡规则,违反则旋转
    6. //空树时
    7. if (_root == nullptr)
    8. {
    9. _root = new Node(kv);
    10. return true;
    11. }
    12. //找到插入位置
    13. Node* parent = nullptr;
    14. Node* cur = _root;
    15. while (cur)
    16. {
    17. if (cur->_kv.first < kv.first)
    18. {
    19. parent = cur;
    20. cur = cur->_right;
    21. }
    22. else if (cur->_kv.first > kv.first)
    23. {
    24. parent = cur;
    25. cur = cur->_left;
    26. }
    27. else
    28. {
    29. return false;
    30. }
    31. }
    32. cur = new Node(kv); //找到结点,创建结点,链接结点
    33. if (parent->_kv.first < kv.first)
    34. {
    35. parent->_right = cur;
    36. }
    37. else
    38. {
    39. parent->_left = cur;
    40. }
    41. cur->_parent = parent; //注意结点的更新
    42. // 已经将结点插入了,现在判断平衡因子--不影响另一颗子树,只会影响祖先
    43. // (新增在父亲右,父亲平衡因子++,新增在左父亲平衡因子--)
    44. //子树高度变了继续往上更新,高度不变停止更新
    45. //没有违反规则,结束,违反规则--旋转处理
    46. while (parent) //依次往上更新,最坏情况下最远更新到根
    47. {
    48. if (cur==parent->_right)
    49. {
    50. parent->_bf++;
    51. }
    52. else
    53. {
    54. parent->_bf--;
    55. }
    56. //是否继续更
    57. if (parent->_bf == 0)
    58. {
    59. break; //高度不变
    60. }
    61. else if (parent->_bf == 1 || parent->_bf == -1)
    62. {
    63. //子树高度变了,继续网上更新祖先
    64. cur = cur->_parent;
    65. parent = parent->_parent;
    66. }
    67. else if (parent->_bf == 2 || parent->_bf == -2)
    68. {
    69. //子树不平衡旋转
    70. if (parent->_bf == 2 && cur->_bf == 1) //左单旋
    71. {
    72. RotateL(parent);
    73. }
    74. else if (parent->_bf == -2 && cur->_bf == -1) //右单旋
    75. {
    76. RotateR(parent);
    77. }
    78. else if (parent->_bf == -2 && cur->_bf == 1) //左右双旋
    79. {
    80. RotateLR(parent);
    81. }
    82. else if (parent->_bf == 2 && cur->_bf == -1) //右左双旋
    83. {
    84. RotateRL(parent);
    85. }
    86. break;
    87. }
    88. else
    89. {
    90. //插入之前AVL就不平衡了
    91. assert(false);
    92. }
    93. }
    94. return true;
    95. }

     

  • 相关阅读:
    Mysql在项目中相关使用(简单操作数据库)
    多项式加法_C语言
    基于STM32+NBIOT(BC26)设计的物联网观赏鱼缸
    图片怎么批量重命名从1到50?
    10474 - Where is the Marble? (UVA)
    使用Vue实现弹窗效果
    Java中快速掌握正则表达式
    脑机接口的发展研究
    div内文字水平居中+垂直居中
    【Dotnet 工具箱】基于 .NET 6 和 Angular 构建项目任务管理平台
  • 原文地址:https://blog.csdn.net/weixin_53316121/article/details/126167732