C++实现AVL树的示例详解

更新时间：2023年03月03日 11:22:17 作者：叫我小秦就好了

AVL Tree 是一个「加上了额外平衡条件」的二叉搜索树，其平衡条件的建立是为了确保整棵树的深度为O(log_2N)，本文主要介绍了AVL树的实现，需要的可以参考一下

AVL 树的概念

也许因为插入的值不够随机，也许因为经过某些插入或删除操作，二叉搜索树可能会失去平衡，甚至可能退化为单链表，造成搜索效率低。

AVL Tree 是一个「加上了额外平衡条件」的二叉搜索树，其平衡条件的建立是为了确保整棵树的深度为 O(log₂N)。

AVL Tree 要求任何节点的左右子树高度相差最多为 1。当违反该规定时，就需要进行旋转来保证该规定。

AVL 树的实现

节点的定义

AVL 树节点的定义比一般的二叉搜索树复杂，它需要额外一个 parent 指针，方便后续旋转。并在每个节点中引入平衡因子，便于判断是否需要旋转。

/// @brief AVL 树节点结构
/// @tparam K 节点的 key 值
/// @tparam V 节点的 value 值
template <class K, class V>
struct AVLTreeNode {
	AVLTreeNode(const pair<K, V>& kv) 
		: _kv(kv)
		, _parent(nullptr)
		, _left(nullptr)
		, _right(nullptr)
		, _bf(0)
	{}

	pair<K, V> _kv;
	AVLTreeNode<K, V>* _parent;
	AVLTreeNode<K, V>* _left;
	AVLTreeNode<K, V>* _right;
    // 左右子树高度相同平衡因子为：0
    // 左子树高平衡因子为负
    // 右子树高平衡因子为正
	int _bf;
};

接口总览

template<class K, class V>
class AVLTree {
	typedef AVLTreeNode<K, V> Node;
public:
	Node* Find(const K& key);
	bool Insert(const pair<K, V>& kv);

private:
	void RotateR(Node* parent);
	void RotateL(Node* parent);
	void RotateLR(Node* parent);
	void RotateRL(Node* parent);
private:
	Node* _root = nullptr;
};

查找

AVL 树的查找和普通的搜索二叉树一样：

若 key 值大于当前节点的值，在当前节点的右子树中查找
若 key 值小于当前节点的值，在当前节点的左子树中查找
若 key 值等于当前节点的值，返回当前节点的地址
若找到空，查找失败，返回空指针

/// @brief 查找指定 key 值
/// @param key 要查找的 key
/// @return 找到返回节点的指针，没找到返回空指针
Node* Find(const K& key) {
    Node* cur = _root;
    while (cur != nullptr) {
        // key 值与当前节点值比较
        if (key > cur->_kv.first) {
            cur = cur->_right;
        } else if (key < cur->_kv.first) {
            cur = cur->_left;
        } else {
            return cur;
        }
    }
    return nullptr;
}

插入

AVL 的插入整体分为两步：

按照二叉搜索树的方式将节点插入
调整节点的平衡因子

平衡因子是怎么调整的？

设新插入的节点为 pCur，新插入节点的父节点为 pParent。在插入之前，pParent 的平衡因子有三种可能：0、-1、1。

插入分为两种：

pCur 插入到 pParent 的左侧，将 pParent 的平衡因子减 1
pCur 插入到 pParent 的右侧，将 pParent 的平衡因子加 1

此时，pParent 的平衡因子可能有三种情况：0、正负 1、正负 2。

0：说明插入之前是正负 1，插入后被调整为 0，满足 AVL 性质插入成功
正负 1：说明插入之前是 0，插入后被调整为正负 1，此时 pParent 变高，需要继续向上更新
正负 2：说明插入之前是正负 1，插入后被调整为正负 2，此时破坏了规定，需要旋转处理

/// @brief 插入指定节点
/// @param kv 待插入的节点
/// @return 插入成功返回 true，失败返回 false
bool Insert(const pair<K, V>& kv) {
    if (_root == nullptr) {
        _root = new Node(kv);
        return true;
    }

    // 先找到要插入的位置
    Node* parent = nullptr;
    Node* cur = _root;
    while (cur != nullptr) {
        if (kv.first > cur->_kv.first) {
            parent = cur;
            cur = cur->_right;
        } else if (kv.first < cur->_kv.first) {
            parent = cur;
            cur = cur->_left;
        } else {
            // 已经存在，插入失败
            return false;
        }
    }

    // 将节点插入
    cur = new Node(kv);
    if (kv.first > parent->_kv.first) {
        parent->_right = cur;
        cur->_parent = parent;
    } else {
        parent->_left = cur;
        cur->_parent = parent;
    }

    // 更新平衡因子，直到正常
    while (parent != nullptr) {
        // 调整父亲的平衡因子
        if (parent->_left == cur) {
            --parent->_bf;
        } else {
            ++parent->_bf;
        }

        if (parent->_bf == 0) {
            // 此时不需要再继续调整了，直接退出
            break;
        } else if (parent->_bf == 1 || parent->_bf == -1) {
            // 此时需要继续向上调整
            cur = parent;
            parent = parent->_parent;
        } else if (parent->_bf == 2 || parent->_bf == -2) {
            // 此时需要旋转处理
            if (parent->_bf == -2 && cur->_bf == -1) {
                RotateR(parent);
            } else if (parent->_bf == 2 && cur->_bf == 1) {
                RotateL(parent);
            } else if (parent->_bf == -2 && cur->_bf == 1) {
                RotateLR(parent);
            } else if (parent->_bf == 2 && cur->_bf == -1) {
                RotateRL(parent);
            } else {
                assert(false);
            }
            // 旋转完了就平衡了，直接退出
            break;
        } else {
            // 此时说明之前就处理错了
            assert(false);
        } // end of if (parent->_bf == 0)
    } // end of while (parent != nullptr)
    return true;
}

旋转

假设平衡因子为正负 2 的节点为 X，由于节点最多拥有两个子节点，因此可以分为四种情况：

插入点位于 X 的左子节点的左子树——左左：右单旋
插入点位于 X 的左子节点的右子树——左右：左右双旋
插入点位于 X 的右子节点的右子树——右右：左单旋
插入点位于 X 的右子节点的左子树——右左：右左双旋

右单旋

假设平衡因子为正负 2 的节点为 parent，parent 的父节点为 pParent，parent 的左子树为 subL，subL 的右子树为 subLR。

右单旋的操作流程：

让 subLR 作为 parent 的左子树
让 parent 作为 subL 的右子树
让 subL 作为整个子树的新根
更新平衡因子

/// @brief 进行右单旋
/// @param parent 平衡因子为正负 2 的节点
void RotateR(Node* parent) {
    Node* pParent = parent->_parent;
    Node* subL = parent->_left;
    Node* subLR = parent->_left->_right;

    // 更改链接关系
    // 1. subLR 作为 parent 的左子树
    parent->_left = subLR;
    if (subLR != nullptr) {
        subLR->_parent = parent;
    }
    // 2. parent 作为 subL 的右子树
    subL->_right = parent;
    parent->_parent = subL;

    // 3. subL 作为整个子树的新根
    if (parent == _root) {
        // parent 为 _root，此时令 subL 为 _root
        _root = subL;
        subL->_parent = nullptr;
    } else {
        // parent 不为 _root，pParent 也就不为空
        if (parent == pParent->_left) {
            pParent->_left = subL;
        } else {
            pParent->_right = subL;
        }
        subL->_parent = pParent;
    }

    // 4. 更新平衡因子
    // 观察上图明显可知
    subL->_bf = 0;
    parent->_bf = 0;
}

左单旋

左单旋与右单旋类似，只是方向不同。

假设平衡因子为正负 2 的节点为 parent，parent 的父节点为 pParent，parent 的右子树为 subR，subR 的左子树为 subRL。

左单旋的操作流程：

让 subRL 作为 parent 的右子树
让 parent 作为 subR 的左子树
让 subR 作为整个子树的新根
更新平衡因子

/// @brief 进行左单旋
/// @param parent 平衡因子为正负 2 的节点
void RotateL(Node* parent) {
    Node* pParetn = parent->_parent;
    Node* subR = parent->_right;
    Node* subRL = parent->_right->_left;

    // 更改链接关系
    // 1. subRL 作为 parent 的右子树
    parent->_right = subRL;
    if (subRL != nullptr) {
        subRL->_parent = parent;
    }
    // 2. parent 作为 subR 的左子树
    subR->_left = parent;
    parent->_parent = subR;

    // 3. subR 作为整个子树的新根
    if (parent == _root) {
        _root = subR;
        subR->_parent = nullptr;
    } else {
        if (parent == pParetn->_left) {
            pParetn->_left = subR;
        } else {
            pParetn->_right = subR;
        }
        subR->_parent = pParetn;
    }

    // 4. 更新平衡因子
    subR->_bf = 0;
    parent->_bf = 0;
}

左右双旋

假设平衡因子为正负 2 的节点为 parent，parent 的左子树为 subL，subL 的右子树为 subLR。

左右双旋就是对 subL 进行一次左单旋，对 parent 进行一次右单旋。双旋也就完成了，要注意的是双旋后平衡因子的更新。

此时分三种情况：

1.新插入的节点是 subLR 的右子树

2.新插入的节点是 subLR 的左子树

3.新插入的是 subLR

结合上述情况，写出如下代码：

/// @brief 进行左右双旋
/// @param parent 平衡因子为正负 2 的节点
void RotateLR(Node* parent) {
    Node* subL = parent->_left;
    Node* subLR = parent->_left->_right;
    int bf = subLR->_bf;

    RotateL(subL);
    RotateR(parent);

    if (bf == 1) {
        // 新插入节点是 subLR 的右子树
        parent->_bf = 0;
        subL->_bf = -1;
        subLR->_bf = 0;
    } else if (bf == -1) {
        // 新插入的节点是 subLR 的左子树
        parent->_bf = 1;
        subL->_bf = 0;
        subLR->_bf = 0;
    } else if (bf == 0) {
        // 新插入的节点是 subLR
        parent->_bf = 0;
        subL->_bf = 0;
        subLR->_bf = 0;
    } else {
        assert(false);
    }
}

右左双旋

假设平衡因子为正负 2 的节点为 parent，parent 的右子树为 subR，subR 的左子树为 subRL。

右左双旋就是对 subR 进行一次右单旋，对 parent 进行一次左单旋。流程和左右双旋一样，这里就不过多介绍了。

void RotateRL(Node* parent) {
    Node* subR = parent->_right;
    Node* subRL = parent->_right->_left;
    int bf = subRL->_bf;

    RotateR(subR);
    RotateL(parent);

    if (bf == 1) {
        // 新插入节点是 subRL 的右子树
        parent->_bf = -1;
        subR->_bf = 0;
        subRL->_bf = 0;
    } else if (bf == -1) {
        // 新插入的节点是 subRL 的左子树
        parent->_bf = 0;
        subR->_bf = 1;
        subRL->_bf = 0;
    } else if (bf == 0) {
        // 新插入的节点是 subRL
        parent->_bf = 0;
        subR->_bf = 0;
        subRL->_bf = 0;
    } else {
        assert(false);
    }
}

以上就是C++实现AVL树的示例详解的详细内容，更多关于C++ AVL树的资料请关注脚本之家其它相关文章！

您可能感兴趣的文章:

C++
AVL树

C++11标准库bind函数应用教程
bind函数定义在头文件functional中，可以将bind函数看做成一个通用的函数适配器，他接收一个可调用对象，生成一个新的可调用对象来"适应"原对象的参数列表。本文将带大家详细了解一下bind函数的应用详解
2021-12-12
C++中的头文件与Extern(外部函数调用)方式
这篇文章主要介绍了C++中的头文件与Extern(外部函数调用)方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
2023-08-08
C语言实现简易通讯录完整流程
这篇文章主要为大家介绍了C语言实现简易通讯录的完整流程，每个环节都有完整代码，有需要的朋友可以借鉴参考下，希望能够有所帮助
2022-02-02
linux c模拟ls命令详解
本篇文章是对linux中基于c模拟ls命令的实现方法进行了详细的分析介绍，需要的朋友参考下
2013-06-06
c语言重要的字符串与内存函数
这篇文章主要介绍一些c语言中常用字符串函数和内存函数的使用和注意事项，并且为了帮助读者理解和使用，也都模拟实现了他们的代码,需要的朋友可以参考一下
2021-09-09
C语言详细分析讲解流程控制语句用法
C语言语句的执行默认顺序执行(从上往下依次执行),编程语言一般除了默认的顺序执行以外，还提供分支执行和循环执行的语法，让我们一起来看看
2022-05-05
C++中虚继承时的构造函数示例详解
在虚继承中,虚基类是由最终的派生类初始化的,换句话说,最终派生类的构造函数必须要调用虚基类的构造函数,这跟普通继承不同,在普通继承中,派生类构造函数中只能调用直接基类的构造函数,不能调用间接基类的,所以本文将通过代码示例给大家介绍一下C++虚继承构造函数
2023-09-09
C++数据序列化方式(自定义结构体的保存和读取)
这篇文章主要介绍了C++数据序列化方式(自定义结构体的保存和读取),具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
2023-08-08
详解C语言中Char型指针数组与字符数组的区别
这篇文章主要介绍了详解C语言中Char型指针数组与字符数组的区别的相关资料,希望通过本文能帮助到大家掌握理解这部分内容，需要的朋友可以参考下
2017-10-10
Qt学习之容器的使用详解
Qt容器主要优点就是在所有的平台上的运行都表现的一致,并且它们都是隐含共享的，这篇文章就来和大家讲讲Qt中容器的具体用法吧，希望对大家有所帮助
2023-03-03

C++实现AVL树的示例详解

目录

AVL 树的概念

AVL 树的实现

节点的定义

接口总览

插入

旋转

相关文章

最新评论

大家感兴趣的内容

最近更新的内容

常用在线小工具