7.5 AVL 樹 *¶
在“二元搜尋樹”章節中我們提到,在多次插入和刪除操作後,二元搜尋樹可能退化為鏈結串列。在這種情況下,所有操作的時間複雜度將從 \(O(\log n)\) 劣化為 \(O(n)\) 。
如圖 7-24 所示,經過兩次刪除節點操作,這棵二元搜尋樹便會退化為鏈結串列。
圖 7-24 AVL 樹在刪除節點後發生退化
再例如,在圖 7-25 所示的完美二元樹中插入兩個節點後,樹將嚴重向左傾斜,查詢操作的時間複雜度也隨之劣化。
圖 7-25 AVL 樹在插入節點後發生退化
1962 年 G. M. Adelson-Velsky 和 E. M. Landis 在論文“An algorithm for the organization of information”中提出了 AVL 樹。論文中詳細描述了一系列操作,確保在持續新增和刪除節點後,AVL 樹不會退化,從而使得各種操作的時間複雜度保持在 \(O(\log n)\) 級別。換句話說,在需要頻繁進行增刪查改操作的場景中,AVL 樹能始終保持高效的資料操作效能,具有很好的應用價值。
7.5.1 AVL 樹常見術語¶
AVL 樹既是二元搜尋樹,也是平衡二元樹,同時滿足這兩類二元樹的所有性質,因此是一種平衡二元搜尋樹(balanced binary search tree)。
1. 節點高度¶
由於 AVL 樹的相關操作需要獲取節點高度,因此我們需要為節點類別新增 height
變數:
/* AVL 樹節點類別 */
class TreeNode {
val; // 節點值
height; //節點高度
left; // 左子節點指標
right; // 右子節點指標
constructor(val, left, right, height) {
this.val = val === undefined ? 0 : val;
this.height = height === undefined ? 0 : height;
this.left = left === undefined ? null : left;
this.right = right === undefined ? null : right;
}
}
/* AVL 樹節點類別 */
class TreeNode {
val: number; // 節點值
height: number; // 節點高度
left: TreeNode | null; // 左子節點指標
right: TreeNode | null; // 右子節點指標
constructor(val?: number, height?: number, left?: TreeNode | null, right?: TreeNode | null) {
this.val = val === undefined ? 0 : val;
this.height = height === undefined ? 0 : height;
this.left = left === undefined ? null : left;
this.right = right === undefined ? null : right;
}
}
use std::rc::Rc;
use std::cell::RefCell;
/* AVL 樹節點結構體 */
struct TreeNode {
val: i32, // 節點值
height: i32, // 節點高度
left: Option<Rc<RefCell<TreeNode>>>, // 左子節點
right: Option<Rc<RefCell<TreeNode>>>, // 右子節點
}
impl TreeNode {
/* 建構子 */
fn new(val: i32) -> Rc<RefCell<Self>> {
Rc::new(RefCell::new(Self {
val,
height: 0,
left: None,
right: None
}))
}
}
/* AVL 樹節點結構體 */
TreeNode struct TreeNode {
int val;
int height;
struct TreeNode *left;
struct TreeNode *right;
} TreeNode;
/* 建構子 */
TreeNode *newTreeNode(int val) {
TreeNode *node;
node = (TreeNode *)malloc(sizeof(TreeNode));
node->val = val;
node->height = 0;
node->left = NULL;
node->right = NULL;
return node;
}
“節點高度”是指從該節點到它的最遠葉節點的距離,即所經過的“邊”的數量。需要特別注意的是,葉節點的高度為 \(0\) ,而空節點的高度為 \(-1\) 。我們將建立兩個工具函式,分別用於獲取和更新節點的高度:
/* 獲取節點高度 */
func (t *aVLTree) height(node *TreeNode) int {
// 空節點高度為 -1 ,葉節點高度為 0
if node != nil {
return node.Height
}
return -1
}
/* 更新節點高度 */
func (t *aVLTree) updateHeight(node *TreeNode) {
lh := t.height(node.Left)
rh := t.height(node.Right)
// 節點高度等於最高子樹高度 + 1
if lh > rh {
node.Height = lh + 1
} else {
node.Height = rh + 1
}
}
/* 獲取節點高度 */
fn height(node: OptionTreeNodeRc) -> i32 {
// 空節點高度為 -1 ,葉節點高度為 0
match node {
Some(node) => node.borrow().height,
None => -1,
}
}
/* 更新節點高度 */
fn update_height(node: OptionTreeNodeRc) {
if let Some(node) = node {
let left = node.borrow().left.clone();
let right = node.borrow().right.clone();
// 節點高度等於最高子樹高度 + 1
node.borrow_mut().height = std::cmp::max(Self::height(left), Self::height(right)) + 1;
}
}
/* 獲取節點高度 */
int height(TreeNode *node) {
// 空節點高度為 -1 ,葉節點高度為 0
if (node != NULL) {
return node->height;
}
return -1;
}
/* 更新節點高度 */
void updateHeight(TreeNode *node) {
int lh = height(node->left);
int rh = height(node->right);
// 節點高度等於最高子樹高度 + 1
if (lh > rh) {
node->height = lh + 1;
} else {
node->height = rh + 1;
}
}
// 獲取節點高度
fn height(self: *Self, node: ?*inc.TreeNode(T)) i32 {
_ = self;
// 空節點高度為 -1 ,葉節點高度為 0
return if (node == null) -1 else node.?.height;
}
// 更新節點高度
fn updateHeight(self: *Self, node: ?*inc.TreeNode(T)) void {
// 節點高度等於最高子樹高度 + 1
node.?.height = @max(self.height(node.?.left), self.height(node.?.right)) + 1;
}
2. 節點平衡因子¶
節點的平衡因子(balance factor)定義為節點左子樹的高度減去右子樹的高度,同時規定空節點的平衡因子為 \(0\) 。我們同樣將獲取節點平衡因子的功能封裝成函式,方便後續使用:
Tip
設平衡因子為 \(f\) ,則一棵 AVL 樹的任意節點的平衡因子皆滿足 \(-1 \le f \le 1\) 。
7.5.2 AVL 樹旋轉¶
AVL 樹的特點在於“旋轉”操作,它能夠在不影響二元樹的中序走訪序列的前提下,使失衡節點重新恢復平衡。換句話說,旋轉操作既能保持“二元搜尋樹”的性質,也能使樹重新變為“平衡二元樹”。
我們將平衡因子絕對值 \(> 1\) 的節點稱為“失衡節點”。根據節點失衡情況的不同,旋轉操作分為四種:右旋、左旋、先右旋後左旋、先左旋後右旋。下面詳細介紹這些旋轉操作。
1. 右旋¶
如圖 7-26 所示,節點下方為平衡因子。從底至頂看,二元樹中首個失衡節點是“節點 3”。我們關注以該失衡節點為根節點的子樹,將該節點記為 node
,其左子節點記為 child
,執行“右旋”操作。完成右旋後,子樹恢復平衡,並且仍然保持二元搜尋樹的性質。
圖 7-26 右旋操作步驟
如圖 7-27 所示,當節點 child
有右子節點(記為 grand_child
)時,需要在右旋中新增一步:將 grand_child
作為 node
的左子節點。
圖 7-27 有 grand_child 的右旋操作
“向右旋轉”是一種形象化的說法,實際上需要透過修改節點指標來實現,程式碼如下所示:
/* 右旋操作 */
fn right_rotate(node: OptionTreeNodeRc) -> OptionTreeNodeRc {
match node {
Some(node) => {
let child = node.borrow().left.clone().unwrap();
let grand_child = child.borrow().right.clone();
// 以 child 為原點,將 node 向右旋轉
child.borrow_mut().right = Some(node.clone());
node.borrow_mut().left = grand_child;
// 更新節點高度
Self::update_height(Some(node));
Self::update_height(Some(child.clone()));
// 返回旋轉後子樹的根節點
Some(child)
}
None => None,
}
}
// 右旋操作
fn rightRotate(self: *Self, node: ?*inc.TreeNode(T)) ?*inc.TreeNode(T) {
var child = node.?.left;
var grandChild = child.?.right;
// 以 child 為原點,將 node 向右旋轉
child.?.right = node;
node.?.left = grandChild;
// 更新節點高度
self.updateHeight(node);
self.updateHeight(child);
// 返回旋轉後子樹的根節點
return child;
}
2. 左旋¶
相應地,如果考慮上述失衡二元樹的“映象”,則需要執行圖 7-28 所示的“左旋”操作。
圖 7-28 左旋操作
同理,如圖 7-29 所示,當節點 child
有左子節點(記為 grand_child
)時,需要在左旋中新增一步:將 grand_child
作為 node
的右子節點。
圖 7-29 有 grand_child 的左旋操作
可以觀察到,右旋和左旋操作在邏輯上是映象對稱的,它們分別解決的兩種失衡情況也是對稱的。基於對稱性,我們只需將右旋的實現程式碼中的所有的 left
替換為 right
,將所有的 right
替換為 left
,即可得到左旋的實現程式碼:
/* 左旋操作 */
fn left_rotate(node: OptionTreeNodeRc) -> OptionTreeNodeRc {
match node {
Some(node) => {
let child = node.borrow().right.clone().unwrap();
let grand_child = child.borrow().left.clone();
// 以 child 為原點,將 node 向左旋轉
child.borrow_mut().left = Some(node.clone());
node.borrow_mut().right = grand_child;
// 更新節點高度
Self::update_height(Some(node));
Self::update_height(Some(child.clone()));
// 返回旋轉後子樹的根節點
Some(child)
}
None => None,
}
}
// 左旋操作
fn leftRotate(self: *Self, node: ?*inc.TreeNode(T)) ?*inc.TreeNode(T) {
var child = node.?.right;
var grandChild = child.?.left;
// 以 child 為原點,將 node 向左旋轉
child.?.left = node;
node.?.right = grandChild;
// 更新節點高度
self.updateHeight(node);
self.updateHeight(child);
// 返回旋轉後子樹的根節點
return child;
}
3. 先左旋後右旋¶
對於圖 7-30 中的失衡節點 3 ,僅使用左旋或右旋都無法使子樹恢復平衡。此時需要先對 child
執行“左旋”,再對 node
執行“右旋”。
圖 7-30 先左旋後右旋
4. 先右旋後左旋¶
如圖 7-31 所示,對於上述失衡二元樹的映象情況,需要先對 child
執行“右旋”,再對 node
執行“左旋”。
圖 7-31 先右旋後左旋
5. 旋轉的選擇¶
圖 7-32 展示的四種失衡情況與上述案例逐個對應,分別需要採用右旋、先左旋後右旋、先右旋後左旋、左旋的操作。
圖 7-32 AVL 樹的四種旋轉情況
如下表所示,我們透過判斷失衡節點的平衡因子以及較高一側子節點的平衡因子的正負號,來確定失衡節點屬於圖 7-32 中的哪種情況。
表 7-3 四種旋轉情況的選擇條件
失衡節點的平衡因子 | 子節點的平衡因子 | 應採用的旋轉方法 |
---|---|---|
\(> 1\) (左偏樹) | \(\geq 0\) | 右旋 |
\(> 1\) (左偏樹) | \(<0\) | 先左旋後右旋 |
\(< -1\) (右偏樹) | \(\leq 0\) | 左旋 |
\(< -1\) (右偏樹) | \(>0\) | 先右旋後左旋 |
為了便於使用,我們將旋轉操作封裝成一個函式。有了這個函式,我們就能對各種失衡情況進行旋轉,使失衡節點重新恢復平衡。程式碼如下所示:
def rotate(self, node: TreeNode | None) -> TreeNode | None:
"""執行旋轉操作,使該子樹重新恢復平衡"""
# 獲取節點 node 的平衡因子
balance_factor = self.balance_factor(node)
# 左偏樹
if balance_factor > 1:
if self.balance_factor(node.left) >= 0:
# 右旋
return self.right_rotate(node)
else:
# 先左旋後右旋
node.left = self.left_rotate(node.left)
return self.right_rotate(node)
# 右偏樹
elif balance_factor < -1:
if self.balance_factor(node.right) <= 0:
# 左旋
return self.left_rotate(node)
else:
# 先右旋後左旋
node.right = self.right_rotate(node.right)
return self.left_rotate(node)
# 平衡樹,無須旋轉,直接返回
return node
/* 執行旋轉操作,使該子樹重新恢復平衡 */
TreeNode *rotate(TreeNode *node) {
// 獲取節點 node 的平衡因子
int _balanceFactor = balanceFactor(node);
// 左偏樹
if (_balanceFactor > 1) {
if (balanceFactor(node->left) >= 0) {
// 右旋
return rightRotate(node);
} else {
// 先左旋後右旋
node->left = leftRotate(node->left);
return rightRotate(node);
}
}
// 右偏樹
if (_balanceFactor < -1) {
if (balanceFactor(node->right) <= 0) {
// 左旋
return leftRotate(node);
} else {
// 先右旋後左旋
node->right = rightRotate(node->right);
return leftRotate(node);
}
}
// 平衡樹,無須旋轉,直接返回
return node;
}
/* 執行旋轉操作,使該子樹重新恢復平衡 */
TreeNode rotate(TreeNode node) {
// 獲取節點 node 的平衡因子
int balanceFactor = balanceFactor(node);
// 左偏樹
if (balanceFactor > 1) {
if (balanceFactor(node.left) >= 0) {
// 右旋
return rightRotate(node);
} else {
// 先左旋後右旋
node.left = leftRotate(node.left);
return rightRotate(node);
}
}
// 右偏樹
if (balanceFactor < -1) {
if (balanceFactor(node.right) <= 0) {
// 左旋
return leftRotate(node);
} else {
// 先右旋後左旋
node.right = rightRotate(node.right);
return leftRotate(node);
}
}
// 平衡樹,無須旋轉,直接返回
return node;
}
/* 執行旋轉操作,使該子樹重新恢復平衡 */
TreeNode? Rotate(TreeNode? node) {
// 獲取節點 node 的平衡因子
int balanceFactorInt = BalanceFactor(node);
// 左偏樹
if (balanceFactorInt > 1) {
if (BalanceFactor(node?.left) >= 0) {
// 右旋
return RightRotate(node);
} else {
// 先左旋後右旋
node!.left = LeftRotate(node!.left);
return RightRotate(node);
}
}
// 右偏樹
if (balanceFactorInt < -1) {
if (BalanceFactor(node?.right) <= 0) {
// 左旋
return LeftRotate(node);
} else {
// 先右旋後左旋
node!.right = RightRotate(node!.right);
return LeftRotate(node);
}
}
// 平衡樹,無須旋轉,直接返回
return node;
}
/* 執行旋轉操作,使該子樹重新恢復平衡 */
func (t *aVLTree) rotate(node *TreeNode) *TreeNode {
// 獲取節點 node 的平衡因子
// Go 推薦短變數,這裡 bf 指代 t.balanceFactor
bf := t.balanceFactor(node)
// 左偏樹
if bf > 1 {
if t.balanceFactor(node.Left) >= 0 {
// 右旋
return t.rightRotate(node)
} else {
// 先左旋後右旋
node.Left = t.leftRotate(node.Left)
return t.rightRotate(node)
}
}
// 右偏樹
if bf < -1 {
if t.balanceFactor(node.Right) <= 0 {
// 左旋
return t.leftRotate(node)
} else {
// 先右旋後左旋
node.Right = t.rightRotate(node.Right)
return t.leftRotate(node)
}
}
// 平衡樹,無須旋轉,直接返回
return node
}
/* 執行旋轉操作,使該子樹重新恢復平衡 */
func rotate(node: TreeNode?) -> TreeNode? {
// 獲取節點 node 的平衡因子
let balanceFactor = balanceFactor(node: node)
// 左偏樹
if balanceFactor > 1 {
if self.balanceFactor(node: node?.left) >= 0 {
// 右旋
return rightRotate(node: node)
} else {
// 先左旋後右旋
node?.left = leftRotate(node: node?.left)
return rightRotate(node: node)
}
}
// 右偏樹
if balanceFactor < -1 {
if self.balanceFactor(node: node?.right) <= 0 {
// 左旋
return leftRotate(node: node)
} else {
// 先右旋後左旋
node?.right = rightRotate(node: node?.right)
return leftRotate(node: node)
}
}
// 平衡樹,無須旋轉,直接返回
return node
}
/* 執行旋轉操作,使該子樹重新恢復平衡 */
#rotate(node) {
// 獲取節點 node 的平衡因子
const balanceFactor = this.balanceFactor(node);
// 左偏樹
if (balanceFactor > 1) {
if (this.balanceFactor(node.left) >= 0) {
// 右旋
return this.#rightRotate(node);
} else {
// 先左旋後右旋
node.left = this.#leftRotate(node.left);
return this.#rightRotate(node);
}
}
// 右偏樹
if (balanceFactor < -1) {
if (this.balanceFactor(node.right) <= 0) {
// 左旋
return this.#leftRotate(node);
} else {
// 先右旋後左旋
node.right = this.#rightRotate(node.right);
return this.#leftRotate(node);
}
}
// 平衡樹,無須旋轉,直接返回
return node;
}
/* 執行旋轉操作,使該子樹重新恢復平衡 */
rotate(node: TreeNode): TreeNode {
// 獲取節點 node 的平衡因子
const balanceFactor = this.balanceFactor(node);
// 左偏樹
if (balanceFactor > 1) {
if (this.balanceFactor(node.left) >= 0) {
// 右旋
return this.rightRotate(node);
} else {
// 先左旋後右旋
node.left = this.leftRotate(node.left);
return this.rightRotate(node);
}
}
// 右偏樹
if (balanceFactor < -1) {
if (this.balanceFactor(node.right) <= 0) {
// 左旋
return this.leftRotate(node);
} else {
// 先右旋後左旋
node.right = this.rightRotate(node.right);
return this.leftRotate(node);
}
}
// 平衡樹,無須旋轉,直接返回
return node;
}
/* 執行旋轉操作,使該子樹重新恢復平衡 */
TreeNode? rotate(TreeNode? node) {
// 獲取節點 node 的平衡因子
int factor = balanceFactor(node);
// 左偏樹
if (factor > 1) {
if (balanceFactor(node!.left) >= 0) {
// 右旋
return rightRotate(node);
} else {
// 先左旋後右旋
node.left = leftRotate(node.left);
return rightRotate(node);
}
}
// 右偏樹
if (factor < -1) {
if (balanceFactor(node!.right) <= 0) {
// 左旋
return leftRotate(node);
} else {
// 先右旋後左旋
node.right = rightRotate(node.right);
return leftRotate(node);
}
}
// 平衡樹,無須旋轉,直接返回
return node;
}
/* 執行旋轉操作,使該子樹重新恢復平衡 */
fn rotate(node: OptionTreeNodeRc) -> OptionTreeNodeRc {
// 獲取節點 node 的平衡因子
let balance_factor = Self::balance_factor(node.clone());
// 左偏樹
if balance_factor > 1 {
let node = node.unwrap();
if Self::balance_factor(node.borrow().left.clone()) >= 0 {
// 右旋
Self::right_rotate(Some(node))
} else {
// 先左旋後右旋
let left = node.borrow().left.clone();
node.borrow_mut().left = Self::left_rotate(left);
Self::right_rotate(Some(node))
}
}
// 右偏樹
else if balance_factor < -1 {
let node = node.unwrap();
if Self::balance_factor(node.borrow().right.clone()) <= 0 {
// 左旋
Self::left_rotate(Some(node))
} else {
// 先右旋後左旋
let right = node.borrow().right.clone();
node.borrow_mut().right = Self::right_rotate(right);
Self::left_rotate(Some(node))
}
} else {
// 平衡樹,無須旋轉,直接返回
node
}
}
/* 執行旋轉操作,使該子樹重新恢復平衡 */
TreeNode *rotate(TreeNode *node) {
// 獲取節點 node 的平衡因子
int bf = balanceFactor(node);
// 左偏樹
if (bf > 1) {
if (balanceFactor(node->left) >= 0) {
// 右旋
return rightRotate(node);
} else {
// 先左旋後右旋
node->left = leftRotate(node->left);
return rightRotate(node);
}
}
// 右偏樹
if (bf < -1) {
if (balanceFactor(node->right) <= 0) {
// 左旋
return leftRotate(node);
} else {
// 先右旋後左旋
node->right = rightRotate(node->right);
return leftRotate(node);
}
}
// 平衡樹,無須旋轉,直接返回
return node;
}
/* 執行旋轉操作,使該子樹重新恢復平衡 */
fun rotate(node: TreeNode): TreeNode {
// 獲取節點 node 的平衡因子
val balanceFactor = balanceFactor(node)
// 左偏樹
if (balanceFactor > 1) {
if (balanceFactor(node.left) >= 0) {
// 右旋
return rightRotate(node)
} else {
// 先左旋後右旋
node.left = leftRotate(node.left)
return rightRotate(node)
}
}
// 右偏樹
if (balanceFactor < -1) {
if (balanceFactor(node.right) <= 0) {
// 左旋
return leftRotate(node)
} else {
// 先右旋後左旋
node.right = rightRotate(node.right)
return leftRotate(node)
}
}
// 平衡樹,無須旋轉,直接返回
return node
}
### 執行旋轉操作,使該子樹重新恢復平衡 ###
def rotate(node)
# 獲取節點 node 的平衡因子
balance_factor = balance_factor(node)
# 左遍樹
if balance_factor > 1
if balance_factor(node.left) >= 0
# 右旋
return right_rotate(node)
else
# 先左旋後右旋
node.left = left_rotate(node.left)
return right_rotate(node)
end
# 右遍樹
elsif balance_factor < -1
if balance_factor(node.right) <= 0
# 左旋
return left_rotate(node)
else
# 先右旋後左旋
node.right = right_rotate(node.right)
return left_rotate(node)
end
end
# 平衡樹,無須旋轉,直接返回
node
end
// 執行旋轉操作,使該子樹重新恢復平衡
fn rotate(self: *Self, node: ?*inc.TreeNode(T)) ?*inc.TreeNode(T) {
// 獲取節點 node 的平衡因子
var balance_factor = self.balanceFactor(node);
// 左偏樹
if (balance_factor > 1) {
if (self.balanceFactor(node.?.left) >= 0) {
// 右旋
return self.rightRotate(node);
} else {
// 先左旋後右旋
node.?.left = self.leftRotate(node.?.left);
return self.rightRotate(node);
}
}
// 右偏樹
if (balance_factor < -1) {
if (self.balanceFactor(node.?.right) <= 0) {
// 左旋
return self.leftRotate(node);
} else {
// 先右旋後左旋
node.?.right = self.rightRotate(node.?.right);
return self.leftRotate(node);
}
}
// 平衡樹,無須旋轉,直接返回
return node;
}
7.5.3 AVL 樹常用操作¶
1. 插入節點¶
AVL 樹的節點插入操作與二元搜尋樹在主體上類似。唯一的區別在於,在 AVL 樹中插入節點後,從該節點到根節點的路徑上可能會出現一系列失衡節點。因此,我們需要從這個節點開始,自底向上執行旋轉操作,使所有失衡節點恢復平衡。程式碼如下所示:
def insert(self, val):
"""插入節點"""
self._root = self.insert_helper(self._root, val)
def insert_helper(self, node: TreeNode | None, val: int) -> TreeNode:
"""遞迴插入節點(輔助方法)"""
if node is None:
return TreeNode(val)
# 1. 查詢插入位置並插入節點
if val < node.val:
node.left = self.insert_helper(node.left, val)
elif val > node.val:
node.right = self.insert_helper(node.right, val)
else:
# 重複節點不插入,直接返回
return node
# 更新節點高度
self.update_height(node)
# 2. 執行旋轉操作,使該子樹重新恢復平衡
return self.rotate(node)
/* 插入節點 */
void insert(int val) {
root = insertHelper(root, val);
}
/* 遞迴插入節點(輔助方法) */
TreeNode *insertHelper(TreeNode *node, int val) {
if (node == nullptr)
return new TreeNode(val);
/* 1. 查詢插入位置並插入節點 */
if (val < node->val)
node->left = insertHelper(node->left, val);
else if (val > node->val)
node->right = insertHelper(node->right, val);
else
return node; // 重複節點不插入,直接返回
updateHeight(node); // 更新節點高度
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = rotate(node);
// 返回子樹的根節點
return node;
}
/* 插入節點 */
void insert(int val) {
root = insertHelper(root, val);
}
/* 遞迴插入節點(輔助方法) */
TreeNode insertHelper(TreeNode node, int val) {
if (node == null)
return new TreeNode(val);
/* 1. 查詢插入位置並插入節點 */
if (val < node.val)
node.left = insertHelper(node.left, val);
else if (val > node.val)
node.right = insertHelper(node.right, val);
else
return node; // 重複節點不插入,直接返回
updateHeight(node); // 更新節點高度
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = rotate(node);
// 返回子樹的根節點
return node;
}
/* 插入節點 */
void Insert(int val) {
root = InsertHelper(root, val);
}
/* 遞迴插入節點(輔助方法) */
TreeNode? InsertHelper(TreeNode? node, int val) {
if (node == null) return new TreeNode(val);
/* 1. 查詢插入位置並插入節點 */
if (val < node.val)
node.left = InsertHelper(node.left, val);
else if (val > node.val)
node.right = InsertHelper(node.right, val);
else
return node; // 重複節點不插入,直接返回
UpdateHeight(node); // 更新節點高度
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = Rotate(node);
// 返回子樹的根節點
return node;
}
/* 插入節點 */
func (t *aVLTree) insert(val int) {
t.root = t.insertHelper(t.root, val)
}
/* 遞迴插入節點(輔助函式) */
func (t *aVLTree) insertHelper(node *TreeNode, val int) *TreeNode {
if node == nil {
return NewTreeNode(val)
}
/* 1. 查詢插入位置並插入節點 */
if val < node.Val.(int) {
node.Left = t.insertHelper(node.Left, val)
} else if val > node.Val.(int) {
node.Right = t.insertHelper(node.Right, val)
} else {
// 重複節點不插入,直接返回
return node
}
// 更新節點高度
t.updateHeight(node)
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = t.rotate(node)
// 返回子樹的根節點
return node
}
/* 插入節點 */
func insert(val: Int) {
root = insertHelper(node: root, val: val)
}
/* 遞迴插入節點(輔助方法) */
func insertHelper(node: TreeNode?, val: Int) -> TreeNode? {
var node = node
if node == nil {
return TreeNode(x: val)
}
/* 1. 查詢插入位置並插入節點 */
if val < node!.val {
node?.left = insertHelper(node: node?.left, val: val)
} else if val > node!.val {
node?.right = insertHelper(node: node?.right, val: val)
} else {
return node // 重複節點不插入,直接返回
}
updateHeight(node: node) // 更新節點高度
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = rotate(node: node)
// 返回子樹的根節點
return node
}
/* 插入節點 */
insert(val) {
this.root = this.#insertHelper(this.root, val);
}
/* 遞迴插入節點(輔助方法) */
#insertHelper(node, val) {
if (node === null) return new TreeNode(val);
/* 1. 查詢插入位置並插入節點 */
if (val < node.val) node.left = this.#insertHelper(node.left, val);
else if (val > node.val)
node.right = this.#insertHelper(node.right, val);
else return node; // 重複節點不插入,直接返回
this.#updateHeight(node); // 更新節點高度
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = this.#rotate(node);
// 返回子樹的根節點
return node;
}
/* 插入節點 */
insert(val: number): void {
this.root = this.insertHelper(this.root, val);
}
/* 遞迴插入節點(輔助方法) */
insertHelper(node: TreeNode, val: number): TreeNode {
if (node === null) return new TreeNode(val);
/* 1. 查詢插入位置並插入節點 */
if (val < node.val) {
node.left = this.insertHelper(node.left, val);
} else if (val > node.val) {
node.right = this.insertHelper(node.right, val);
} else {
return node; // 重複節點不插入,直接返回
}
this.updateHeight(node); // 更新節點高度
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = this.rotate(node);
// 返回子樹的根節點
return node;
}
/* 插入節點 */
void insert(int val) {
root = insertHelper(root, val);
}
/* 遞迴插入節點(輔助方法) */
TreeNode? insertHelper(TreeNode? node, int val) {
if (node == null) return TreeNode(val);
/* 1. 查詢插入位置並插入節點 */
if (val < node.val)
node.left = insertHelper(node.left, val);
else if (val > node.val)
node.right = insertHelper(node.right, val);
else
return node; // 重複節點不插入,直接返回
updateHeight(node); // 更新節點高度
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = rotate(node);
// 返回子樹的根節點
return node;
}
/* 插入節點 */
fn insert(&mut self, val: i32) {
self.root = Self::insert_helper(self.root.clone(), val);
}
/* 遞迴插入節點(輔助方法) */
fn insert_helper(node: OptionTreeNodeRc, val: i32) -> OptionTreeNodeRc {
match node {
Some(mut node) => {
/* 1. 查詢插入位置並插入節點 */
match {
let node_val = node.borrow().val;
node_val
}
.cmp(&val)
{
Ordering::Greater => {
let left = node.borrow().left.clone();
node.borrow_mut().left = Self::insert_helper(left, val);
}
Ordering::Less => {
let right = node.borrow().right.clone();
node.borrow_mut().right = Self::insert_helper(right, val);
}
Ordering::Equal => {
return Some(node); // 重複節點不插入,直接返回
}
}
Self::update_height(Some(node.clone())); // 更新節點高度
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = Self::rotate(Some(node)).unwrap();
// 返回子樹的根節點
Some(node)
}
None => Some(TreeNode::new(val)),
}
}
/* 插入節點 */
void insert(AVLTree *tree, int val) {
tree->root = insertHelper(tree->root, val);
}
/* 遞迴插入節點(輔助函式) */
TreeNode *insertHelper(TreeNode *node, int val) {
if (node == NULL) {
return newTreeNode(val);
}
/* 1. 查詢插入位置並插入節點 */
if (val < node->val) {
node->left = insertHelper(node->left, val);
} else if (val > node->val) {
node->right = insertHelper(node->right, val);
} else {
// 重複節點不插入,直接返回
return node;
}
// 更新節點高度
updateHeight(node);
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = rotate(node);
// 返回子樹的根節點
return node;
}
/* 插入節點 */
fun insert(_val: Int) {
root = insertHelper(root, _val)
}
/* 遞迴插入節點(輔助方法) */
fun insertHelper(n: TreeNode?, _val: Int): TreeNode {
if (n == null)
return TreeNode(_val)
var node = n
/* 1. 查詢插入位置並插入節點 */
if (_val < node._val)
node.left = insertHelper(node.left, _val)
else if (_val > node._val)
node.right = insertHelper(node.right, _val)
else
return node // 重複節點不插入,直接返回
updateHeight(node) // 更新節點高度
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = rotate(node)
// 返回子樹的根節點
return node
}
### 插入節點 ###
def insert(val)
@root = insert_helper(@root, val)
end
### 遞迴插入節點(輔助方法)###
def insert_helper(node, val)
return TreeNode.new(val) if node.nil?
# 1. 查詢插入位置並插入節點
if val < node.val
node.left = insert_helper(node.left, val)
elsif val > node.val
node.right = insert_helper(node.right, val)
else
# 重複節點不插入,直接返回
return node
end
# 更新節點高度
update_height(node)
# 2. 執行旋轉操作,使該子樹重新恢復平衡
rotate(node)
end
// 插入節點
fn insert(self: *Self, val: T) !void {
self.root = (try self.insertHelper(self.root, val)).?;
}
// 遞迴插入節點(輔助方法)
fn insertHelper(self: *Self, node_: ?*inc.TreeNode(T), val: T) !?*inc.TreeNode(T) {
var node = node_;
if (node == null) {
var tmp_node = try self.mem_allocator.create(inc.TreeNode(T));
tmp_node.init(val);
return tmp_node;
}
// 1. 查詢插入位置並插入節點
if (val < node.?.val) {
node.?.left = try self.insertHelper(node.?.left, val);
} else if (val > node.?.val) {
node.?.right = try self.insertHelper(node.?.right, val);
} else {
return node; // 重複節點不插入,直接返回
}
self.updateHeight(node); // 更新節點高度
// 2. 執行旋轉操作,使該子樹重新恢復平衡
node = self.rotate(node);
// 返回子樹的根節點
return node;
}
2. 刪除節點¶
類似地,在二元搜尋樹的刪除節點方法的基礎上,需要從底至頂執行旋轉操作,使所有失衡節點恢復平衡。程式碼如下所示:
def remove(self, val: int):
"""刪除節點"""
self._root = self.remove_helper(self._root, val)
def remove_helper(self, node: TreeNode | None, val: int) -> TreeNode | None:
"""遞迴刪除節點(輔助方法)"""
if node is None:
return None
# 1. 查詢節點並刪除
if val < node.val:
node.left = self.remove_helper(node.left, val)
elif val > node.val:
node.right = self.remove_helper(node.right, val)
else:
if node.left is None or node.right is None:
child = node.left or node.right
# 子節點數量 = 0 ,直接刪除 node 並返回
if child is None:
return None
# 子節點數量 = 1 ,直接刪除 node
else:
node = child
else:
# 子節點數量 = 2 ,則將中序走訪的下個節點刪除,並用該節點替換當前節點
temp = node.right
while temp.left is not None:
temp = temp.left
node.right = self.remove_helper(node.right, temp.val)
node.val = temp.val
# 更新節點高度
self.update_height(node)
# 2. 執行旋轉操作,使該子樹重新恢復平衡
return self.rotate(node)
/* 刪除節點 */
void remove(int val) {
root = removeHelper(root, val);
}
/* 遞迴刪除節點(輔助方法) */
TreeNode *removeHelper(TreeNode *node, int val) {
if (node == nullptr)
return nullptr;
/* 1. 查詢節點並刪除 */
if (val < node->val)
node->left = removeHelper(node->left, val);
else if (val > node->val)
node->right = removeHelper(node->right, val);
else {
if (node->left == nullptr || node->right == nullptr) {
TreeNode *child = node->left != nullptr ? node->left : node->right;
// 子節點數量 = 0 ,直接刪除 node 並返回
if (child == nullptr) {
delete node;
return nullptr;
}
// 子節點數量 = 1 ,直接刪除 node
else {
delete node;
node = child;
}
} else {
// 子節點數量 = 2 ,則將中序走訪的下個節點刪除,並用該節點替換當前節點
TreeNode *temp = node->right;
while (temp->left != nullptr) {
temp = temp->left;
}
int tempVal = temp->val;
node->right = removeHelper(node->right, temp->val);
node->val = tempVal;
}
}
updateHeight(node); // 更新節點高度
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = rotate(node);
// 返回子樹的根節點
return node;
}
/* 刪除節點 */
void remove(int val) {
root = removeHelper(root, val);
}
/* 遞迴刪除節點(輔助方法) */
TreeNode removeHelper(TreeNode node, int val) {
if (node == null)
return null;
/* 1. 查詢節點並刪除 */
if (val < node.val)
node.left = removeHelper(node.left, val);
else if (val > node.val)
node.right = removeHelper(node.right, val);
else {
if (node.left == null || node.right == null) {
TreeNode child = node.left != null ? node.left : node.right;
// 子節點數量 = 0 ,直接刪除 node 並返回
if (child == null)
return null;
// 子節點數量 = 1 ,直接刪除 node
else
node = child;
} else {
// 子節點數量 = 2 ,則將中序走訪的下個節點刪除,並用該節點替換當前節點
TreeNode temp = node.right;
while (temp.left != null) {
temp = temp.left;
}
node.right = removeHelper(node.right, temp.val);
node.val = temp.val;
}
}
updateHeight(node); // 更新節點高度
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = rotate(node);
// 返回子樹的根節點
return node;
}
/* 刪除節點 */
void Remove(int val) {
root = RemoveHelper(root, val);
}
/* 遞迴刪除節點(輔助方法) */
TreeNode? RemoveHelper(TreeNode? node, int val) {
if (node == null) return null;
/* 1. 查詢節點並刪除 */
if (val < node.val)
node.left = RemoveHelper(node.left, val);
else if (val > node.val)
node.right = RemoveHelper(node.right, val);
else {
if (node.left == null || node.right == null) {
TreeNode? child = node.left ?? node.right;
// 子節點數量 = 0 ,直接刪除 node 並返回
if (child == null)
return null;
// 子節點數量 = 1 ,直接刪除 node
else
node = child;
} else {
// 子節點數量 = 2 ,則將中序走訪的下個節點刪除,並用該節點替換當前節點
TreeNode? temp = node.right;
while (temp.left != null) {
temp = temp.left;
}
node.right = RemoveHelper(node.right, temp.val!.Value);
node.val = temp.val;
}
}
UpdateHeight(node); // 更新節點高度
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = Rotate(node);
// 返回子樹的根節點
return node;
}
/* 刪除節點 */
func (t *aVLTree) remove(val int) {
t.root = t.removeHelper(t.root, val)
}
/* 遞迴刪除節點(輔助函式) */
func (t *aVLTree) removeHelper(node *TreeNode, val int) *TreeNode {
if node == nil {
return nil
}
/* 1. 查詢節點並刪除 */
if val < node.Val.(int) {
node.Left = t.removeHelper(node.Left, val)
} else if val > node.Val.(int) {
node.Right = t.removeHelper(node.Right, val)
} else {
if node.Left == nil || node.Right == nil {
child := node.Left
if node.Right != nil {
child = node.Right
}
if child == nil {
// 子節點數量 = 0 ,直接刪除 node 並返回
return nil
} else {
// 子節點數量 = 1 ,直接刪除 node
node = child
}
} else {
// 子節點數量 = 2 ,則將中序走訪的下個節點刪除,並用該節點替換當前節點
temp := node.Right
for temp.Left != nil {
temp = temp.Left
}
node.Right = t.removeHelper(node.Right, temp.Val.(int))
node.Val = temp.Val
}
}
// 更新節點高度
t.updateHeight(node)
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = t.rotate(node)
// 返回子樹的根節點
return node
}
/* 刪除節點 */
func remove(val: Int) {
root = removeHelper(node: root, val: val)
}
/* 遞迴刪除節點(輔助方法) */
func removeHelper(node: TreeNode?, val: Int) -> TreeNode? {
var node = node
if node == nil {
return nil
}
/* 1. 查詢節點並刪除 */
if val < node!.val {
node?.left = removeHelper(node: node?.left, val: val)
} else if val > node!.val {
node?.right = removeHelper(node: node?.right, val: val)
} else {
if node?.left == nil || node?.right == nil {
let child = node?.left ?? node?.right
// 子節點數量 = 0 ,直接刪除 node 並返回
if child == nil {
return nil
}
// 子節點數量 = 1 ,直接刪除 node
else {
node = child
}
} else {
// 子節點數量 = 2 ,則將中序走訪的下個節點刪除,並用該節點替換當前節點
var temp = node?.right
while temp?.left != nil {
temp = temp?.left
}
node?.right = removeHelper(node: node?.right, val: temp!.val)
node?.val = temp!.val
}
}
updateHeight(node: node) // 更新節點高度
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = rotate(node: node)
// 返回子樹的根節點
return node
}
/* 刪除節點 */
remove(val) {
this.root = this.#removeHelper(this.root, val);
}
/* 遞迴刪除節點(輔助方法) */
#removeHelper(node, val) {
if (node === null) return null;
/* 1. 查詢節點並刪除 */
if (val < node.val) node.left = this.#removeHelper(node.left, val);
else if (val > node.val)
node.right = this.#removeHelper(node.right, val);
else {
if (node.left === null || node.right === null) {
const child = node.left !== null ? node.left : node.right;
// 子節點數量 = 0 ,直接刪除 node 並返回
if (child === null) return null;
// 子節點數量 = 1 ,直接刪除 node
else node = child;
} else {
// 子節點數量 = 2 ,則將中序走訪的下個節點刪除,並用該節點替換當前節點
let temp = node.right;
while (temp.left !== null) {
temp = temp.left;
}
node.right = this.#removeHelper(node.right, temp.val);
node.val = temp.val;
}
}
this.#updateHeight(node); // 更新節點高度
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = this.#rotate(node);
// 返回子樹的根節點
return node;
}
/* 刪除節點 */
remove(val: number): void {
this.root = this.removeHelper(this.root, val);
}
/* 遞迴刪除節點(輔助方法) */
removeHelper(node: TreeNode, val: number): TreeNode {
if (node === null) return null;
/* 1. 查詢節點並刪除 */
if (val < node.val) {
node.left = this.removeHelper(node.left, val);
} else if (val > node.val) {
node.right = this.removeHelper(node.right, val);
} else {
if (node.left === null || node.right === null) {
const child = node.left !== null ? node.left : node.right;
// 子節點數量 = 0 ,直接刪除 node 並返回
if (child === null) {
return null;
} else {
// 子節點數量 = 1 ,直接刪除 node
node = child;
}
} else {
// 子節點數量 = 2 ,則將中序走訪的下個節點刪除,並用該節點替換當前節點
let temp = node.right;
while (temp.left !== null) {
temp = temp.left;
}
node.right = this.removeHelper(node.right, temp.val);
node.val = temp.val;
}
}
this.updateHeight(node); // 更新節點高度
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = this.rotate(node);
// 返回子樹的根節點
return node;
}
/* 刪除節點 */
void remove(int val) {
root = removeHelper(root, val);
}
/* 遞迴刪除節點(輔助方法) */
TreeNode? removeHelper(TreeNode? node, int val) {
if (node == null) return null;
/* 1. 查詢節點並刪除 */
if (val < node.val)
node.left = removeHelper(node.left, val);
else if (val > node.val)
node.right = removeHelper(node.right, val);
else {
if (node.left == null || node.right == null) {
TreeNode? child = node.left ?? node.right;
// 子節點數量 = 0 ,直接刪除 node 並返回
if (child == null)
return null;
// 子節點數量 = 1 ,直接刪除 node
else
node = child;
} else {
// 子節點數量 = 2 ,則將中序走訪的下個節點刪除,並用該節點替換當前節點
TreeNode? temp = node.right;
while (temp!.left != null) {
temp = temp.left;
}
node.right = removeHelper(node.right, temp.val);
node.val = temp.val;
}
}
updateHeight(node); // 更新節點高度
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = rotate(node);
// 返回子樹的根節點
return node;
}
/* 刪除節點 */
fn remove(&self, val: i32) {
Self::remove_helper(self.root.clone(), val);
}
/* 遞迴刪除節點(輔助方法) */
fn remove_helper(node: OptionTreeNodeRc, val: i32) -> OptionTreeNodeRc {
match node {
Some(mut node) => {
/* 1. 查詢節點並刪除 */
if val < node.borrow().val {
let left = node.borrow().left.clone();
node.borrow_mut().left = Self::remove_helper(left, val);
} else if val > node.borrow().val {
let right = node.borrow().right.clone();
node.borrow_mut().right = Self::remove_helper(right, val);
} else if node.borrow().left.is_none() || node.borrow().right.is_none() {
let child = if node.borrow().left.is_some() {
node.borrow().left.clone()
} else {
node.borrow().right.clone()
};
match child {
// 子節點數量 = 0 ,直接刪除 node 並返回
None => {
return None;
}
// 子節點數量 = 1 ,直接刪除 node
Some(child) => node = child,
}
} else {
// 子節點數量 = 2 ,則將中序走訪的下個節點刪除,並用該節點替換當前節點
let mut temp = node.borrow().right.clone().unwrap();
loop {
let temp_left = temp.borrow().left.clone();
if temp_left.is_none() {
break;
}
temp = temp_left.unwrap();
}
let right = node.borrow().right.clone();
node.borrow_mut().right = Self::remove_helper(right, temp.borrow().val);
node.borrow_mut().val = temp.borrow().val;
}
Self::update_height(Some(node.clone())); // 更新節點高度
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = Self::rotate(Some(node)).unwrap();
// 返回子樹的根節點
Some(node)
}
None => None,
}
}
/* 刪除節點 */
// 由於引入了 stdio.h ,此處無法使用 remove 關鍵詞
void removeItem(AVLTree *tree, int val) {
TreeNode *root = removeHelper(tree->root, val);
}
/* 遞迴刪除節點(輔助函式) */
TreeNode *removeHelper(TreeNode *node, int val) {
TreeNode *child, *grandChild;
if (node == NULL) {
return NULL;
}
/* 1. 查詢節點並刪除 */
if (val < node->val) {
node->left = removeHelper(node->left, val);
} else if (val > node->val) {
node->right = removeHelper(node->right, val);
} else {
if (node->left == NULL || node->right == NULL) {
child = node->left;
if (node->right != NULL) {
child = node->right;
}
// 子節點數量 = 0 ,直接刪除 node 並返回
if (child == NULL) {
return NULL;
} else {
// 子節點數量 = 1 ,直接刪除 node
node = child;
}
} else {
// 子節點數量 = 2 ,則將中序走訪的下個節點刪除,並用該節點替換當前節點
TreeNode *temp = node->right;
while (temp->left != NULL) {
temp = temp->left;
}
int tempVal = temp->val;
node->right = removeHelper(node->right, temp->val);
node->val = tempVal;
}
}
// 更新節點高度
updateHeight(node);
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = rotate(node);
// 返回子樹的根節點
return node;
}
/* 刪除節點 */
fun remove(_val: Int) {
root = removeHelper(root, _val)
}
/* 遞迴刪除節點(輔助方法) */
fun removeHelper(n: TreeNode?, _val: Int): TreeNode? {
var node = n ?: return null
/* 1. 查詢節點並刪除 */
if (_val < node._val)
node.left = removeHelper(node.left, _val)
else if (_val > node._val)
node.right = removeHelper(node.right, _val)
else {
if (node.left == null || node.right == null) {
val child = if (node.left != null)
node.left
else
node.right
// 子節點數量 = 0 ,直接刪除 node 並返回
if (child == null)
return null
// 子節點數量 = 1 ,直接刪除 node
else
node = child
} else {
// 子節點數量 = 2 ,則將中序走訪的下個節點刪除,並用該節點替換當前節點
var temp = node.right
while (temp!!.left != null) {
temp = temp.left
}
node.right = removeHelper(node.right, temp._val)
node._val = temp._val
}
}
updateHeight(node) // 更新節點高度
/* 2. 執行旋轉操作,使該子樹重新恢復平衡 */
node = rotate(node)
// 返回子樹的根節點
return node
}
### 刪除節點 ###
def remove(val)
@root = remove_helper(@root, val)
end
### 遞迴刪除節點(輔助方法)###
def remove_helper(node, val)
return if node.nil?
# 1. 查詢節點並刪除
if val < node.val
node.left = remove_helper(node.left, val)
elsif val > node.val
node.right = remove_helper(node.right, val)
else
if node.left.nil? || node.right.nil?
child = node.left || node.right
# 子節點數量 = 0 ,直接刪除 node 並返回
return if child.nil?
# 子節點數量 = 1 ,直接刪除 node
node = child
else
# 子節點數量 = 2 ,則將中序走訪的下個節點刪除,並用該節點替換當前節點
temp = node.right
while !temp.left.nil?
temp = temp.left
end
node.right = remove_helper(node.right, temp.val)
node.val = temp.val
end
end
# 更新節點高度
update_height(node)
# 2. 執行旋轉操作,使該子樹重新恢復平衡
rotate(node)
end
// 刪除節點
fn remove(self: *Self, val: T) void {
self.root = self.removeHelper(self.root, val).?;
}
// 遞迴刪除節點(輔助方法)
fn removeHelper(self: *Self, node_: ?*inc.TreeNode(T), val: T) ?*inc.TreeNode(T) {
var node = node_;
if (node == null) return null;
// 1. 查詢節點並刪除
if (val < node.?.val) {
node.?.left = self.removeHelper(node.?.left, val);
} else if (val > node.?.val) {
node.?.right = self.removeHelper(node.?.right, val);
} else {
if (node.?.left == null or node.?.right == null) {
var child = if (node.?.left != null) node.?.left else node.?.right;
// 子節點數量 = 0 ,直接刪除 node 並返回
if (child == null) {
return null;
// 子節點數量 = 1 ,直接刪除 node
} else {
node = child;
}
} else {
// 子節點數量 = 2 ,則將中序走訪的下個節點刪除,並用該節點替換當前節點
var temp = node.?.right;
while (temp.?.left != null) {
temp = temp.?.left;
}
node.?.right = self.removeHelper(node.?.right, temp.?.val);
node.?.val = temp.?.val;
}
}
self.updateHeight(node); // 更新節點高度
// 2. 執行旋轉操作,使該子樹重新恢復平衡
node = self.rotate(node);
// 返回子樹的根節點
return node;
}
3. 查詢節點¶
AVL 樹的節點查詢操作與二元搜尋樹一致,在此不再贅述。
7.5.4 AVL 樹典型應用¶
- 組織和儲存大型資料,適用於高頻查詢、低頻增刪的場景。
- 用於構建資料庫中的索引系統。
- 紅黑樹也是一種常見的平衡二元搜尋樹。相較於 AVL 樹,紅黑樹的平衡條件更寬鬆,插入與刪除節點所需的旋轉操作更少,節點增刪操作的平均效率更高。