[C++刷怪笼]:AVL树--平衡二叉查找树的先驱

发布于:2025-09-10 ⋅ 阅读:(15) ⋅ 点赞:(0)

目录

1.AVL树的概念

2.AVL树的实现

2.1AVL树的框架实现

2.2AVL树的插入(insert)

2.2.1插入的过程

2.2.2平衡因子的更新

2.2.3插入节点及更新平衡因子的代码实现

2.3AVL树的旋转(rotate)

2.3.1旋转的原则

2.3.2右单旋(RotateR)

2.3.4左单旋(RotateL)

2.3.5左右双旋(RotateLR)

2.3.6右左双旋

2.4AVL树的查找(find)

2.5AVL树平衡检测

3.对AVL树的总结

1.AVL树的概念

AVL树是最先发明的⾃平衡⼆叉查找树,AVL是⼀颗空树,或者具备下列性质的⼆叉搜索树:它的
左右⼦树都是AVL树,且左右⼦树的⾼度差的绝对值不超过1。AVL树是⼀颗⾼度平衡搜索⼆叉树,通过控制⾼度差去控制平衡。
AVL树得名于它的发明者G. M. Adelson-Velsky和E. M. Landis是两个前苏联的科学家,他们在1962年的论⽂《An algorithm for the organization of information》中发表了它。
AVL树实现这⾥我们引⼊⼀个平衡因⼦(balance factor)的概念,每个结点都有⼀个平衡因⼦,任何
结点的平衡因⼦等于右⼦树的⾼度减去左⼦树的⾼度,也就是说任何结点的平衡因⼦等于0/1/-1,
AVL树并不是必须要平衡因⼦,但是有了平衡因⼦可以更⽅便我们去进⾏观察和控制树是否平衡,
就像⼀个⻛向标⼀样。
思考⼀下为什么AVL树是⾼度平衡搜索⼆叉树,要求⾼度差不超过1,⽽不是⾼度差是0呢?0不是更好的平衡吗?画画图分析我们发现,不是不想这样设计,⽽是有些情况是做不到⾼度差是0的。⽐如⼀棵树是2个结点,4个结点等情况下,⾼度差最好就是1,⽆法做到⾼度差是0
AVL树整体结点数量和分布和完全⼆叉树类似,⾼度可以控制在logN,那么增删查改的效率也可
以控制在O ( logN ),相⽐⼆叉搜索树有了本质的提升。
下图就是一个非常经典的AVL树,目前不用去在意节点顶上的数字,后面我们会详细的去解释。

2.AVL树的实现

2.1AVL树的框架实现

#include<iostream>
#include<map>
#include<assert.h>
#include<vector>
using namespace std;

template<class K, class V>
struct AVLTNode
{
    /* data */
public:
    AVLTNode(cosnt pair<K, V> kv)
        : _kv(kv)
        , _left(nullptr)
        , _right(nullptr)
        , _parent(nullptr)
        , _bf(0)
    {}
private:
    pair<K, V> _kv;
    AVLTNode<K, V>* _left;
    AVLTNode<K, V>* _right;
    AVLTNode<K, V>* _parent;
    int _bf;//banlance factor(右子树 - 左子树)

};

template<class K, class V>
class AVLTree
{
    typedef AVLTreeNode<K, V> Node;
public:
    //...
private:
    Node* _root = nullptr;
};

上面实现的AVL树的基本框架,接下来我们就来对其进行功能填充--各个接口逐一模拟实现。

2.2AVL树的插入(insert)

2.2.1插入的过程

1. 插⼊⼀个值按⼆叉搜索树规则进⾏插⼊。
2. 新增结点以后,只会影响祖先结点的⾼度,也就是可能会影响部分祖先结点的平衡因⼦,所以更新从新增结点->根结点路径上的平衡因⼦,实际中最坏情况下要更新到根,有些情况更新到中间就可以停⽌了,具体情况我们下⾯再详细分析。
3. 更新平衡因⼦过程中没有出现问题,则插⼊结束
4. 更新平衡因⼦过程中出现不平衡,对不平衡⼦树旋转,旋转后本质调平衡的同时,本质降低了⼦树的⾼度,不会再影响上⼀层,所以插⼊结束。

2.2.2平衡因子的更新

更新原则:

平衡因⼦ = 右⼦树⾼度-左⼦树⾼度。

只有⼦树⾼度变化才会影响当前结点平衡因⼦。

插⼊结点,会增加⾼度,所以新增结点在parent的右⼦树,parent的平衡因⼦++,新增结点在
parent的左⼦树,parent平衡因⼦--。
parent所在⼦树的⾼度是否变化决定了是否会继续往上更新。
更新停止条件:
•  更新后parent的平衡因⼦等于0,更新中parent的平衡因⼦变化为-1->0 或者 1->0,说明更新前
parent⼦树⼀边⾼⼀边低,新增的结点插⼊在低的那边,插⼊后parent所在的⼦树⾼度不变,不会
影响parent的⽗亲结点的平衡因⼦,更新结束。
更新后parent的平衡因⼦等于1 或 -1,更新前更新中parent的平衡因⼦变化为0->1 或者 0->-1,说明更新前parent⼦树两边⼀样⾼,新增的插⼊结点后,parent所在的⼦树⼀边⾼⼀边低,parent所在的⼦树符合平衡要求,但是⾼度增加了1,会影响parent的⽗亲结点的平衡因⼦,所以要继续向上更新。
更新后parent的平衡因⼦等于2 或 -2,更新前更新中parent的平衡因⼦变化为1->2 或者 -1->-2,说 明更新前parent⼦树⼀边⾼⼀边低,新增的插⼊结点在⾼的那边,parent所在的⼦树⾼的那边更⾼ 了,破坏了平衡,parent所在的⼦树不符合平衡要求,需要旋转处理,旋转的⽬标有两个:1、把 parent⼦树旋转平衡。2、降低parent⼦树的⾼度,恢复到插⼊结点以前的⾼度。所以旋转后也不 需要继续往上更新,插⼊结束。
•  不断更新,更新到根,跟的平衡因⼦是1或-1也停⽌了。
更新到10结点,平衡因⼦为2,10所在的⼦树已经不平衡,需要旋转处理

更新到中间结点,3为根的⼦树⾼度不变,不会影响上⼀层,更新结束
最坏更新到根停⽌

2.2.3插入节点及更新平衡因子的代码实现

//插入节点
    bool Insert(const pair<K, V> &kv)
    {
        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;
        }

        //更新平衡因子
        while(parent)
        {
            //插入的节点为父节点的右孩子
            if(parent -> _right == cur)
            {
                parent -> _bf++;
            }
            //插入节点为父节点的左孩子
            else
            {
                parent -> _bf--;
            }

            //更新结束 _bf = 0
            if(parent -> _bf == 0)
                break;
            //继续更新 _bf = 1 || -1
            else if (parent -> _bf == 1 || parent -> _bf == -1)
            {
                //继续向上更新_bf
                cur = parent;
                parent = parent -> _parent;
            }
            //旋转 _bf = -2 || 2
            else if(parent -> _bf == 2 || parent -> _bf == -2)
            {
                //已经不平衡了--旋转处理
                break;
            }
            else
                assert(false);
        }
        return true;
    }

2.3AVL树的旋转(rotate)

2.3.1旋转的原则

1. 保持搜索树的规则
2. 让旋转的树从不满⾜变平衡,其次降低旋转树的⾼度
旋转总共分为四种,左单旋/右单旋/左右双旋/右左双旋。
说明:下⾯的图中,有些结点我们给的是具体值,如10和5等结点,这⾥是为了⽅便讲解,实际中是什么值都可以,只要⼤⼩关系符合搜索树的性质即可。

2.3.2右单旋(RotateR)

本图1展⽰的是10为根的树,有a/b/c抽象为三棵⾼度为h的⼦树(h>=0),a/b/c均符合AVL树的要
求。10可能是整棵树的根,也可能是⼀个整棵树中局部的⼦树的根。这⾥a/b/c是⾼度为h的⼦树,
是⼀种概括抽象表⽰,他代表了所有右单旋的场景,实际右单旋形态有很多种,具体图2/图3/图4/
图5进⾏了详细描述。
在a⼦树中插⼊⼀个新结点,导致a⼦树的⾼度从h变成h+1,不断向上更新平衡因⼦,导致10的平
衡因⼦从-1变成-2,10为根的树左右⾼度差超过1,违反平衡规则。10为根的树左边太⾼了,需要
往右边旋转,控制两棵树的平衡。
旋转核⼼步骤,因为5 < b⼦树的值 < 10,将b变成10的左⼦树,10变成5的右⼦树,5变成这棵树新的根,符合搜索树的规则,控制了平衡,同时这棵的⾼度恢复到了插⼊之前的h+2,符合旋转原
则。如果插⼊之前10整棵树的⼀个局部⼦树,旋转后不会再影响上⼀层,插⼊结束了。
代码实现:
//右单旋
    void RotateR(Node* parent)
    {
        Node* SubL = parent -> _left;
        Node* SubLR = SubL -> _right;

        //除了修改孩子的指向还要修改父亲的指向
        parent -> _left = SubLR;

        //节点为空就可以不用管
        if(SubLR)
            subLR->_parent = parent;
        
        Node* pparent = parent -> _parent;
        SubL -> _right = parent;
        parent -> _parent = SubL;

        // parent有可能是整棵树的根,也可能是局部的⼦树
        // 如果是整棵树的根,要修改_root
        // 如果是局部的指针要跟上⼀层链接
        if(pparent == nullptr)
        {
            _root = SubL;
            SubL -> _parent = nullptr;
        }
        else
        {
            if(parent == pparent -> _left)
            {
                pparent -> _left = SubL;
            }
            else
            {
                pparent -> _right = SubL;
            }
            SubL -> _parent = pparent;
        }
        parent -> _bf = SubL -> _bf = 0;
    }

2.3.4左单旋(RotateL)

本图6展⽰的是10为根的树,有a/b/c抽象为三棵⾼度为h的⼦树(h>=0),a/b/c均符合AVL树的要
求。10可能是整棵树的根,也可能是⼀个整棵树中局部的⼦树的根。这⾥a/b/c是⾼度为h的⼦树,
是⼀种概括抽象表⽰,他代表了所有右单旋的场景,实际右单旋形态有很多种,具体跟上⾯左旋类
似。
在a⼦树中插⼊⼀个新结点,导致a⼦树的⾼度从h变成h+1,不断向上更新平衡因⼦,导致10的平
衡因⼦从1变成2,10为根的树左右⾼度差超过1,违反平衡规则。10为根的树右边太⾼了,需要往
左边旋转,控制两棵树的平衡。
旋转核⼼步骤,因为10 < b⼦树的值 < 15,将b变成10的右⼦树,10变成15的左⼦树,15变成这棵树新的根,符合搜索树的规则,控制了平衡,同时这棵的⾼度恢复到了插⼊之前的h+2,符合旋转原则。如果插⼊之前10整棵树的⼀个局部⼦树,旋转后不会再影响上⼀层,插⼊结束了。
代码实现:
//左单旋
    void RotateL(Node* parent)
    {
        Node* SubR = parent -> _right;
        Node* SubRL = SubR -> _left;

        parent -> _right = SubRL;
        if(SubRL)
            SubRL -> _parent = parent;

        Node* pparent = parent -> _parent;
        SubR -> _left = parent;
        parent -> _parent = SubR;
        if(pparent == nullptr)
        {
            _root = SubR;
            SubR -> _parent = nullptr;
        }
        else
        {
            if(parent == pparent -> _left)
            {
                SubR = pparent -> _left;
            }
            else
            {
                SubR = pparent -> _right;
            }
            SubR -> _parent = pparent;
        }

        SubR -> _bf = parent ->_bf = 0;
    }

2.3.5左右双旋(RotateLR)

可以看到,左边⾼时,如果插⼊位置不是在a⼦树,⽽是插⼊在b⼦树,b⼦树⾼度从h变
成h+1,引发旋转,右单旋⽆法解决问题,右单旋后,我们的树依旧不平衡。右单旋解决的纯粹的左边⾼,但是插⼊在b⼦树中,10为跟的⼦树不再是单纯的左边⾼,对于10是左边⾼,但是对于5是右边⾼,需要⽤两次旋转才能解决,以5为旋转点进⾏⼀个左单旋,以10为旋转点进⾏⼀个右单旋,这棵树这棵树就平衡了。
图7和图8分别为左右双旋中h==0和h==1具体场景分析,下⾯我们将a/b/c⼦树抽象为⾼度h的AVL
⼦树进⾏分析,另外我们需要把b⼦树的细节进⼀步展开为8和左⼦树⾼度为h-1的e和f⼦树,因为
我们要对b的⽗亲5为旋转点进⾏左单旋,左单旋需要动b树中的左⼦树。b⼦树中新增结点的位置
不同,平衡因⼦更新的细节也不同,通过观察8的平衡因⼦不同,这⾥我们要分三个场景讨论。
场景1:h >= 1时,新增结点插⼊在e⼦树,e⼦树⾼度从h-1并为h并不断更新8->5->10平衡因⼦,
引发旋转,其中8的平衡因⼦为-1,旋转后8和5平衡因⼦为0,10平衡因⼦为1。
场景2:h >= 1时,新增结点插⼊在f⼦树,f⼦树⾼度从h-1变为h并不断更新8->5->10平衡因⼦,引发旋转,其中8的平衡因⼦为1,旋转后8和10平衡因⼦为0,5平衡因⼦为-1。
场景3:h == 0时,a/b/c都是空树,b⾃⼰就是⼀个新增结点,不断更新5->10平衡因⼦,引发旋
转,其中8的平衡因⼦为0,旋转后8和10和5平衡因⼦均为0。
代码实现:
//左右双旋
    void RotateLR(Node* parent)
    {
        Node* SubL = parent -> _ left;
        Node* SubLR = SubL -> _right;
        int bf = SubLR -> _bf;

        RotateL(SubL);
        RoutateR(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.3.6右左双旋

跟左右双旋类似,下⾯我们将a/b/c⼦树抽象为⾼度h的AVL⼦树进⾏分析,另外我们需要把b⼦树的细节进⼀步展开为12和左⼦树⾼度为h-1的e和f⼦树,因为我们要对b的⽗亲15为旋转点进⾏右单
旋,右单旋需要动b树中的右⼦树。b⼦树中新增结点的位置不同,平衡因⼦更新的细节也不同,通
过观察12的平衡因⼦不同,这⾥我们要分三个场景讨论。
场景1:h >= 1时,新增结点插⼊在e⼦树,e⼦树⾼度从h-1变为h并不断更新12->15->10平衡因
⼦,引发旋转,其中12的平衡因⼦为-1,旋转后10和12平衡因⼦为0,15平衡因⼦为1。
场景2:h >= 1时,新增结点插⼊在f⼦树,f⼦树⾼度从h-1变为h并不断更新12->15->10平衡因⼦,引发旋转,其中12的平衡因⼦为1,旋转后15和12平衡因⼦为0,10平衡因⼦为-1。
场景3:h == 0时,a/b/c都是空树,b⾃⼰就是⼀个新增结点,不断更新15->10平衡因⼦,引发旋
转,其中12的平衡因⼦为0,旋转后10和12和15平衡因⼦均为0。
代码实现:
    //右左双旋
    void Rotate(Node* parent)
    {
        Node* SubR = parent -> _right;
        Node* SubRL = SubR -> _left;
        int bf = SubRL -> _bf;

        RotateR(SubR);
        RotateL(parent);

        if(bf == 0)
        {
            parent -> _bf = 0;
            SubL -> _bf = 0;
            SubLR -> _bf = 0;
        }
        if(bf == 1)
        {
            parent -> _bf = -1;
            SubR -> _bf = 0;
            SubRL -> _bf = 0;
        }
        if(bf == -1)
        {
            parent -> _bf = 0;
            SubRL -> _bf = 0;
            SubR -> _bf = 1
        }
    }

2.4AVL树的查找(find)

⼆叉搜索树逻辑实现即可,搜索效率为 O ( logN )
    Node* Find(const K& key)
    {
        Node* cur = _root;
        while(cur)
        {
            if(cur -> _kv.first < key)
            {
                cur = cur -> right;
            }
            else if(cur -> kv.first > key)
            {
                cur = cur -> left;
            }
            else return cur;
        }
        return nullptr;
    }

2.5AVL树平衡检测

我们实现的AVL树是否合格,我们通过检查左右⼦树⾼度差的的程序进⾏反向验证,同时检查⼀下结点的平衡因⼦更新是否出现了问题。
    int Height(Node* root)
    {
        if(root == nullptr) return 0;

        int leftheight = Height(root -> _left);
        int rightheight = Height(root -> _right);

        return leftheight > rightheight : leftheight + 1 : rightheight + 1;
    }

    bool IsBalanceTree(Node* root)
    {
        if(root == nullptr) return true;

        int leftheight = Height(root -> _left);
        int rightheight = Height(root -> _right);
        int diff = rightheight - leftheight;

        if(abs(diff) >= 2)
        {
            cout << "height diff error" << endl;
            return false;
        }
        if(root -> _bf != diff)
        {
            cout << "banlance factor error" <<endl;
            return false;
        }

        return IsBanlance(root -> _left) && IsBanlance(root -> _right);
    } 

3.对AVL树的总结

AVL 树是一种自平衡二叉搜索树,它在二叉搜索树的基础上,通过特定的机制保证树的高度始终维持在一个相对平衡的状态,即任意节点的左右子树高度差的绝对值不超过 1。这个特性使得 AVL 树在进行插入、删除等操作后,依然能够保持较好的查找性能,时间复杂度稳定在 O(logn),其中 n 是树中节点的数量。

总体而言,AVL 树以其良好的平衡性和稳定的操作复杂度,在众多需要高效数据处理的领域发挥着重要作用,尽管存在一定的缺点,但在合适的应用场景下能展现出明显的优势。


网站公告

今日签到

点亮在社区的每一天
去签到