|
| 1 | +--- |
| 2 | +title: "深入理解并实现基本的平衡二叉树(Balanced Binary Tree)数据结构" |
| 3 | +author: "黄京" |
| 4 | +date: "Sep 16, 2025" |
| 5 | +description: "AVL 树实现与平衡操作详解" |
| 6 | +latex: true |
| 7 | +pdf: true |
| 8 | +--- |
| 9 | + |
| 10 | +> 本文将深入探讨平衡二叉树的核心概念,重点解析为何需要它、它是如何通过旋转操作维持平衡的。我们将以最经典的 AVL 树为例,逐步拆解其四大旋转操作,并最终用代码(C++)完整实现一个具备插入、删除和查询功能的 AVL 树数据结构。无论你是正在学习数据结构的学生,还是希望巩固基础的开发者,这篇文章都将为你提供清晰的指引。 |
| 11 | +
|
| 12 | + |
| 13 | +二叉搜索树(BST)是一种常见的数据结构,它具有高效的搜索、插入和删除操作,理想情况下的时间复杂度为 $O(h)$,其中 $h$ 是树的高度。BST 的定义基于每个节点的值大于其左子树的所有值且小于其右子树的所有值。然而,BST 有一个致命的缺陷:当插入的数据是有序的,例如序列 1, 2, 3, 4, 5,BST 会退化成一条链表,此时高度 $h$ 等于节点数 $n$,操作时间复杂度降至 $O(n)$,效率急剧下降。 |
| 14 | + |
| 15 | +为了解决这个问题,平衡二叉树应运而生。平衡二叉树的核心思想是在 BST 的基础上,通过某些策略控制树的高度,使其尽可能保持「矮胖」的形状,从而保证操作效率始终维持在 $O(\logn)$。常见的平衡树类型包括 AVL 树、红黑树和 Treap 等。本文将聚焦于最基础且易于理解的 AVL 树,因为它提供了严格的平衡保证和相对简单的实现方式。 |
| 16 | + |
| 17 | +## AVL 树:定义与核心概念 |
| 18 | + |
| 19 | +AVL 树是一种自平衡的二叉搜索树,由 Adelson-Velsky 和 Landis 在 1962 年提出。它首先满足 BST 的所有性质,但额外要求每个节点的平衡因子(Balance Factor, BF)必须保持在 -1、0 或 1 的范围内。平衡因子定义为该节点左子树的高度减去右子树的高度,即 $BF = h_{\text{left}} - h_{\text{right}}$。如果任何节点的平衡因子超出这个范围,树就被认为是不平衡的,需要通过旋转操作来调整。 |
| 20 | + |
| 21 | +计算节点高度是 AVL 树操作的基础。通常,我们约定空节点的高度为 -1(这便于计算),而非空节点的高度为其左右子树高度的最大值加一。例如,一个叶子节点的高度为 0,因为它没有子节点。在代码中,我们会频繁调用一个辅助函数来获取或更新节点的高度,以确保平衡因子的正确计算。 |
| 22 | + |
| 23 | +## 维持平衡的魔法:AVL 旋转 |
| 24 | + |
| 25 | +当插入或删除节点导致平衡因子超出允许范围时,AVL 树通过旋转操作来恢复平衡。旋转的目的是在不破坏 BST 排序性质的前提下,对局部子树进行重组,以降低高度。旋转操作分为四种基本类型,对应不同的不平衡情况。 |
| 26 | + |
| 27 | +第一种情况是左左(LL)情况。这发生在节点的平衡因子大于 1(即左子树比右子树高太多),且其左孩子的平衡因子大于或等于 0(表示不平衡源于左孩子的左子树)。解决方案是执行右旋操作:以该节点为轴,将其左孩子提升为新的根节点,原节点成为新根节点的右孩子,同时调整子树指针以维持 BST 性质。 |
| 28 | + |
| 29 | +第二种情况是右右(RR)情况。这发生在节点的平衡因子小于 -1(即右子树比左子树高太多),且其右孩子的平衡因子小于或等于 0(表示不平衡源于右孩子的右子树)。解决方案是执行左旋操作:以该节点为轴,将其右孩子提升为新的根节点,原节点成为新根节点的左孩子,并调整子树指针。 |
| 30 | + |
| 31 | +第三种情况是左右(LR)情况。这发生在节点的平衡因子大于 1,但其左孩子的平衡因子小于 0(表示不平衡源于左孩子的右子树)。解决方案需要两步:先对左孩子执行左旋操作(将其转化为 LL 情况),再对原节点执行右旋操作。 |
| 32 | + |
| 33 | +第四种情况是右左(RL)情况。这发生在节点的平衡因子小于 -1,但其右孩子的平衡因子大于 0(表示不平衡源于右孩子的左子树)。解决方案同样需要两步:先对右孩子执行右旋操作(将其转化为 RR 情况),再对原节点执行左旋操作。 |
| 34 | + |
| 35 | +理解这些旋转的关键在于可视化子树的重组过程,但本文避免使用图片,因此建议读者在脑海中模拟指针的调整。旋转操作完成后,树的高度会减少,平衡因子恢复正常,从而确保整体效率。 |
| 36 | + |
| 37 | +## 代码实现:手把手构建 AVL 树 |
| 38 | + |
| 39 | +我们将使用 C++ 语言实现 AVL 树。代码实现分为多个步骤,从定义节点类到实现核心操作函数。每个代码块后会有详细解读,以帮助理解。 |
| 40 | + |
| 41 | +首先,定义树节点类。节点包含整数值 `val`、左右子节点指针 `left` 和 `right`,以及高度 `height`。我们选择存储高度而非平衡因子,因为平衡因子可以通过高度计算得出。 |
| 42 | + |
| 43 | +```cpp |
| 44 | +class TreeNode { |
| 45 | +public: |
| 46 | + int val; |
| 47 | + TreeNode* left; |
| 48 | + TreeNode* right; |
| 49 | + int height; |
| 50 | + TreeNode(int x) : val(x), left(nullptr), right(nullptr), height(0) {} |
| 51 | +}; |
| 52 | +``` |
| 53 | +
|
| 54 | +这段代码定义了一个简单的节点类,构造函数初始化值、指针和高度。高度初始化为 0,但后续会通过更新函数调整。 |
| 55 | +
|
| 56 | +接下来,实现辅助函数。`getHeight` 函数用于获取节点高度,处理空节点情况;`updateHeight` 函数更新节点高度基于其子节点高度;`getBalanceFactor` 函数计算平衡因子。 |
| 57 | +
|
| 58 | +```cpp |
| 59 | +int getHeight(TreeNode* node) { |
| 60 | + if (node == nullptr) return -1; |
| 61 | + return node->height; |
| 62 | +} |
| 63 | +
|
| 64 | +void updateHeight(TreeNode* node) { |
| 65 | + if (node == nullptr) return; |
| 66 | + node->height = 1 + std::max(getHeight(node->left), getHeight(node->right)); |
| 67 | +} |
| 68 | +
|
| 69 | +int getBalanceFactor(TreeNode* node) { |
| 70 | + if (node == nullptr) return 0; |
| 71 | + return getHeight(node->left) - getHeight(node->right); |
| 72 | +} |
| 73 | +``` |
| 74 | + |
| 75 | +`getHeight` 函数检查节点是否为空,如果是则返回 -1,否则返回存储的高度。`updateHeight` 函数重新计算节点高度为左右子树高度的最大值加一。`getBalanceFactor` 函数返回左子树高度减右子树高度,直接使用高度值计算。 |
| 76 | + |
| 77 | +现在,实现旋转函数。左旋和右旋是基本操作,它们返回调整后的新根节点。 |
| 78 | + |
| 79 | +```cpp |
| 80 | +TreeNode* rotateLeft(TreeNode* y) { |
| 81 | + TreeNode* x = y->right; |
| 82 | + TreeNode* T2 = x->left; |
| 83 | + x->left = y; |
| 84 | + y->right = T2; |
| 85 | + updateHeight(y); |
| 86 | + updateHeight(x); |
| 87 | + return x; |
| 88 | +} |
| 89 | + |
| 90 | +TreeNode* rotateRight(TreeNode* x) { |
| 91 | + TreeNode* y = x->left; |
| 92 | + TreeNode* T2 = y->right; |
| 93 | + y->right = x; |
| 94 | + x->left = T2; |
| 95 | + updateHeight(x); |
| 96 | + updateHeight(y); |
| 97 | + return y; |
| 98 | +} |
| 99 | +``` |
| 100 | +
|
| 101 | +在 `rotateLeft` 函数中,`y` 是原根节点,`x` 是其右孩子。操作将 `x` 的左子树(T2) attached 到 `y` 的右边,然后更新 `y` 和 `x` 的高度。右旋类似,但方向相反。旋转后,BST 性质保持不变,因为值的相对顺序没有改变。 |
| 102 | +
|
| 103 | +基于旋转函数,实现平衡函数 `balance`。它检查节点的平衡因子,并根据四种情况调用相应的旋转。 |
| 104 | +
|
| 105 | +```cpp |
| 106 | +TreeNode* balance(TreeNode* node) { |
| 107 | + if (node == nullptr) return nullptr; |
| 108 | + int bf = getBalanceFactor(node); |
| 109 | + if (bf > 1) { |
| 110 | + if (getBalanceFactor(node->left) >= 0) { |
| 111 | + return rotateRight(node); // LL case |
| 112 | + } else { |
| 113 | + node->left = rotateLeft(node->left); // LR case: first left rotate left child |
| 114 | + return rotateRight(node); |
| 115 | + } |
| 116 | + } |
| 117 | + if (bf < -1) { |
| 118 | + if (getBalanceFactor(node->right) <= 0) { |
| 119 | + return rotateLeft(node); // RR case |
| 120 | + } else { |
| 121 | + node->right = rotateRight(node->right); // RL case: first right rotate right child |
| 122 | + return rotateLeft(node); |
| 123 | + } |
| 124 | + } |
| 125 | + return node; // no need to balance |
| 126 | +} |
| 127 | +``` |
| 128 | + |
| 129 | +这个函数首先计算平衡因子。如果平衡因子大于 1,检查左孩子的平衡因子以区分 LL 或 LR 情况,并执行相应旋转。类似地处理平衡因子小于 -1 的情况。旋转后返回新根节点。 |
| 130 | + |
| 131 | +实现插入操作。插入是递归的,先执行标准 BST 插入,然后更新高度并平衡节点。 |
| 132 | + |
| 133 | +```cpp |
| 134 | +TreeNode* insert(TreeNode* node, int key) { |
| 135 | + if (node == nullptr) { |
| 136 | + return new TreeNode(key); |
| 137 | + } |
| 138 | + if (key < node->val) { |
| 139 | + node->left = insert(node->left, key); |
| 140 | + } else if (key > node->val) { |
| 141 | + node->right = insert(node->right, key); |
| 142 | + } else { |
| 143 | + return node; // duplicate keys not allowed |
| 144 | + } |
| 145 | + updateHeight(node); |
| 146 | + return balance(node); |
| 147 | +} |
| 148 | +``` |
| 149 | +
|
| 150 | +插入函数递归地找到合适位置插入新节点。插入后,更新当前节点高度并调用 `balance` 来恢复平衡。返回调整后的根节点。 |
| 151 | +
|
| 152 | +实现删除操作。删除同样递归,处理三种情况:无子节点、有一个子节点、有两个子节点。删除后更新高度并平衡。 |
| 153 | +
|
| 154 | +```cpp |
| 155 | +TreeNode* remove(TreeNode* node, int key) { |
| 156 | + if (node == nullptr) return nullptr; |
| 157 | + if (key < node->val) { |
| 158 | + node->left = remove(node->left, key); |
| 159 | + } else if (key > node->val) { |
| 160 | + node->right = remove(node->right, key); |
| 161 | + } else { |
| 162 | + if (node->left == nullptr || node->right == nullptr) { |
| 163 | + TreeNode* temp = node->left ? node->left : node->right; |
| 164 | + delete node; |
| 165 | + return temp; |
| 166 | + } else { |
| 167 | + TreeNode* temp = node->right; |
| 168 | + while (temp->left != nullptr) { |
| 169 | + temp = temp->left; |
| 170 | + } |
| 171 | + node->val = temp->val; |
| 172 | + node->right = remove(node->right, temp->val); |
| 173 | + } |
| 174 | + } |
| 175 | + updateHeight(node); |
| 176 | + return balance(node); |
| 177 | +} |
| 178 | +``` |
| 179 | + |
| 180 | +删除函数首先找到要删除的节点。如果节点有一个或无子节点,直接删除并返回子节点。如果有两个子节点,找到中序遍历后继(右子树的最小值),复制值到当前节点,并递归删除后继节点。最后更新高度并平衡。 |
| 181 | + |
| 182 | +查找操作与普通 BST 相同,无需修改。 |
| 183 | + |
| 184 | +```cpp |
| 185 | +bool search(TreeNode* node, int key) { |
| 186 | + if (node == nullptr) return false; |
| 187 | + if (key == node->val) return true; |
| 188 | + if (key < node->val) return search(node->left, key); |
| 189 | + return search(node->right, key); |
| 190 | +} |
| 191 | +``` |
| 192 | +
|
| 193 | +查找函数递归搜索值,返回是否存在。 |
| 194 | +
|
| 195 | +## 测试与验证 |
| 196 | +
|
| 197 | +为了验证 AVL 树的正确性,我们编写一个简单的测试程序。测试数据使用有序序列 [10, 20, 30, 40, 50, 25] 来演示 AVL 树如何避免退化。 |
| 198 | +
|
| 199 | +```cpp |
| 200 | +#include <iostream> |
| 201 | +#include <algorithm> |
| 202 | +void inOrder(TreeNode* node) { |
| 203 | + if (node == nullptr) return; |
| 204 | + inOrder(node->left); |
| 205 | + std::cout << node->val << " "; |
| 206 | + inOrder(node->right); |
| 207 | +} |
| 208 | +int main() { |
| 209 | + TreeNode* root = nullptr; |
| 210 | + root = insert(root, 10); |
| 211 | + root = insert(root, 20); |
| 212 | + root = insert(root, 30); |
| 213 | + root = insert(root, 40); |
| 214 | + root = insert(root, 50); |
| 215 | + root = insert(root, 25); |
| 216 | + std::cout << "In-order traversal: "; |
| 217 | + inOrder(root); // should output sorted values |
| 218 | + std::cout << "\nHeight of tree: " << getHeight(root) << std::endl; // should be small |
| 219 | + // Test deletion if implemented |
| 220 | + root = remove(root, 30); |
| 221 | + std::cout << "After deletion, in-order: "; |
| 222 | + inOrder(root); |
| 223 | + std::cout << std::endl; |
| 224 | + return 0; |
| 225 | +} |
| 226 | +``` |
| 227 | + |
| 228 | +中序遍历应输出有序序列(10, 20, 25, 30, 40, 50),证明 BST 性质维持。树高度应远小于节点数,表示平衡。删除操作后,树应保持平衡和有序。 |
| 229 | + |
| 230 | + |
| 231 | +AVL 树通过严格的平衡条件确保了操作的时间复杂度为 $O(\logn)$,非常适合读多写少的场景。然而,其缺点在于插入和删除时需要频繁旋转,可能导致性能开销。相比之下,红黑树采用近似平衡,旋转次数较少,广泛应用于标准库如 C++ STL 的 map 和 set。 |
| 232 | + |
| 233 | +对于进一步学习,建议探索红黑树、B 树和 Splay 树等其他平衡数据结构。这些结构在不同场景下各有优势,例如 B 树用于数据库和文件系统。 |
| 234 | + |
| 235 | +## 附录 & 互动 |
| 236 | + |
| 237 | +完整代码可参考 GitHub 仓库示例。欢迎在评论区提问或分享您的实现经验。参考文献包括经典算法书籍和在线资源,如 Introduction to Algorithms by Cormen et al.。 |
0 commit comments