c++AVL树（也称高度平衡搜索二叉树）

梦静泽

已于 2023-05-12 17:38:37 修改

阅读量103

点赞数

文章标签： c++ 算法数据结构

于 2023-05-12 17:37:59 首次发布

本文链接：https://blog.csdn.net/weixin_68256194/article/details/130646165

版权

1.新节点插入较高左子树的左侧 --- 左左：右单旋

2.新节点插入较高右子树的右侧 --- 右右：左单旋

————————————————————————————

AVL树（也称高度平衡搜索二叉树）

1.AVL树的概念

二叉搜索树虽可以缩短查找的效率，但如果数据有序或接近有序二叉搜索树将退化为单支树，查找元素相当于在顺序表中搜索元素，效率低下。因此，两位俄罗斯的数学家G.M.Adelson-Velskii和E.M.Landis在1962年发明了一种解决上述问题的方法：当向二叉搜索树中插入新结点后，如果能保证每个结点的左右子树高度之差的绝对值不超过1(需要对树中的结点进行调整)，即可降低树的高度，从而减少平均搜索长度。

一棵AVL树或者是空树，或者是具有以下性质的二叉搜索树：

它的左右子树都是AVL树
左右子树高度之差(简称平衡因子)的绝对值不超过1(-1/0/1)

注意：AVL树不一定有平衡因子，使用平衡因子只是它的一种实现方式。

2.AVL树节点的定义

注：这里使用了平衡因子。

template<class K, class V>
class AVLTreeNode
{
    AVLTreeNode<K, V>* _left;
    AVLTreeNode<K, V>* _right;
    AVLTreeNode<K, V>* _parent;
    pair<K, V> _kv;
    int _bf; // 平衡因子   
    
    AVLTreeNode(const pair<K,V>& kv)
         : _pLeft(nullptr), _pRight(nullptr), _pParent(nullptr)
         , _data(data), _bf(0)
         {}              
};

3.AVL树节点的插入

1、先完成二叉搜索树的插入，插入节点只会影响它到根节点的路径上的祖先。

2、调整平衡因子

如果更新之后，平衡因子没有出现问题 ( |bf| ，平衡结构没有受到影响，不需要处理
如果更新之后，平衡因子出现问题 ( |bf| > 1)，平衡结构受到影响，需要处理(旋转)

新增节点的处理

1、新增节点在父节点的右边

cur == parent->right parent->bf++

2、新增节点在父节点的左边

cur == parent->left parent->bf--

什么决定是否要继续往上更新对应爷爷节点？

取决于parent所在的子树的高度是否发生了变化；如果发生了变化就继续更新，不变则不再更新。

a、parent->bf == 1 || parent->bf == -1 -》 parent所在的子树高度发生改变，继续更新，因为插入之前parent->bf == 0, 说明插入之前左右两边高度相等，现在有一边高，说明parent一边高一边低，高度发生改变

b、parent->bf == 2 || parent->bf == -2 -》parent所在子树不平衡，停止更新，并处理该颗子树（旋转）

c、parent->bf == 0 -》parent所在子树高度不变，停止更新，结束本次插入。

说明插入前是parent->bf == 1 or -1，插入前一边高一边低，插入节点填补矮的那部分节点，使其保持平衡

旋转

旋转的原则：保持它继续是搜索树

旋转的目的：左右均衡，降低整颗树的高度

1.新节点插入较高左子树的左侧 --- 左左：右单旋

右单旋的旋转过程：

1、a变成20的左边

2、20变成40的右边

3、40变成整颗树的根

void RotateR(Node* parent)
{
    Node* subL = parent->_left;
    Node* subLR = subLR->_right;

    Node* grand = parent->_parent;
    parent->_left = subRL;

     if (subRL)
         subRL->_parent = parent;

    subL->_right = parent;
    parent->_parent = subL;

     if (grand == nullptr)
     {
    _root = subL;
    subL->_parent = nullptr;
     }
     else
     {
          if (grand->_left == parent)
            grand->_left = subL;
          else
            grand->right = subL;

          subL->_parent = grand;
     }
     parent->_bf = subL->_bf = 0;
}

2.新节点插入较高右子树的右侧 --- 右右：左单旋

void RotateL(Node* parent)
{
    Node* subR = parent->_right;
    subRL = subR->_left;
    
    parent->_right = subRL;
    if(subRL)
    subRL->parent = parent;
    
    Node* ppnode = parent->_parent;
    
    subR->left = parent;
    parent->_parent = subR;
    
    if(ppnode == nullptr)
    {
        _root = subR;
        _root->_parent = nullptr;    
    }
    else
    {
         if(ppnode->_left == parent)
         {
             ppnode->_left = subR;         
         }
         else if(ppnode->_right == parent)
         {
             ppnode->_right = subR;         
         }   
    } 
    ppnode->_bf = subR->_bf = 0;
}

左右双旋：

复用左单旋和右单旋

void RotateLR(Node* parent)
{
    Node* subL = parent->_left;
    Node* subLR = subL->_rigth;
    int bf = subLR->_bf;
    
    RotateL(parent->_left);
    RotateR(parent);
    
    if(bf == 1)
    {
        parent->_bf = 0;    
    }
}