引言
在計算機科學領域,數據結構是構建高效算法的基石。而在眾多的數據結構中,平衡二叉搜索樹因其優秀的查找、插入和刪除性能而備受關注。紅黑樹(Red-Black Tree)作為一種自平衡的二叉搜索樹,更是在 C++ 標準庫(如 STL 中的 map 和 set)中得到了廣泛應用。本文將深入探討紅黑樹的原理、實現及應用,幫助讀者全面掌握這一重要的數據結構。
紅黑樹的基本概念
紅黑樹是一種特殊的二叉搜索樹,它在每個節點上增加了一個存儲位來表示節點的顏色(紅色或黑色)。通過對任何一條從根到葉子的路徑上各個節點著色方式的限制,紅黑樹確保沒有一條路徑會比其他路徑長出兩倍,因而是接近平衡的。
紅黑樹必須滿足以下五個性質:
- 每個節點要么是紅色,要么是黑色。
- 根節點是黑色。
- 每個葉子節點(NIL 節點,空節點)是黑色的。
- 如果一個節點是紅色的,則它的兩個子節點都是黑色的。
- 對每個節點,從該節點到其所有后代葉節點的簡單路徑上,均包含相同數目的黑色節點。
這些性質的約束使得紅黑樹的高度始終保持在 O (log n),從而保證了基本操作的高效性。
紅黑樹的操作
紅黑樹的基本操作包括插入、刪除和查找。由于紅黑樹是二叉搜索樹的一種,其查找操作與普通二叉搜索樹相同,但插入和刪除操作后需要通過旋轉和變色來維持紅黑樹的性質。
插入操作
插入操作首先按照二叉搜索樹的方式插入新節點,并將其著色為紅色。然后通過一系列的旋轉和顏色調整來修復紅黑樹的性質。插入操作的時間復雜度為 O (log n)。
插入修復主要分為三種情況:
- 情況 1:插入節點的父節點是紅色,且叔叔節點(父節點的兄弟節點)也是紅色。此時將父節點和叔叔節點變為黑色,祖父節點變為紅色,然后繼續處理祖父節點。
- 情況 2:插入節點的父節點是紅色,叔叔節點是黑色,且插入節點是父節點的右孩子。此時進行左旋操作,將問題轉化為情況 3。
- 情況 3:插入節點的父節點是紅色,叔叔節點是黑色,且插入節點是父節點的左孩子。此時將父節點變為黑色,祖父節點變為紅色,然后進行右旋操作。
刪除操作
刪除操作同樣首先按照二叉搜索樹的方式刪除節點,然后通過旋轉和變色來修復紅黑樹的性質。刪除操作的時間復雜度也為 O (log n)。
刪除修復比插入修復更為復雜,主要分為四種情況,需要考慮兄弟節點的顏色以及兄弟節點子節點的顏色。
紅黑樹的 C++ 實現
下面是紅黑樹的 C++ 實現代碼,包括節點結構、插入、刪除、查找等基本操作:
#include "red_black_tree.hpp"
#include <iostream>
#include <cassert>int main() {RedBlackTree<int> tree;// 測試插入操作tree.insert(10);tree.insert(20);tree.insert(5);tree.insert(15);tree.insert(30);std::cout << "Inorder traversal after insertion: ";tree.inorder(); // 應輸出 5 10 15 20 30// 測試查找操作assert(tree.contains(15) == true);assert(tree.contains(25) == false);std::cout << "Contains test passed." << std::endl;// 測試刪除操作tree.remove(20);std::cout << "Inorder traversal after deleting 20: ";tree.inorder(); // 應輸出 5 10 15 30assert(tree.contains(20) == false);// 測試大小assert(tree.size() == 4);std::cout << "Size after deletion: " << tree.size() << std::endl;// 測試空樹RedBlackTree<int> emptyTree;assert(emptyTree.empty() == true);std::cout << "Empty tree test passed." << std::endl;std::cout << "All tests passed!" << std::endl;return 0;
}
#ifndef RED_BLACK_TREE_HPP
#define RED_BLACK_TREE_HPP#include <iostream>
#include <memory>
#include <functional>
#include <cassert>template<typename T, typename Compare = std::less<T>>
class RedBlackTree {
private:enum class Color { RED, BLACK };struct Node {T data;Color color;std::unique_ptr<Node> left;std::unique_ptr<Node> right;Node* parent;Node(const T& value, Color nodeColor = Color::RED): data(value), color(nodeColor), left(nullptr), right(nullptr), parent(nullptr) {}};std::unique_ptr<Node> root;Compare compare;size_t treeSize;// 左旋操作void leftRotate(Node* x) {Node* y = x->right.get();x->right = std::move(y->left);if (y->left) {y->left->parent = x;}y->parent = x->parent;if (!x->parent) {root = std::unique_ptr<Node>(y);} else if (x == x->parent->left.get()) {x->parent->left = std::unique_ptr<Node>(y);} else {x->parent->right = std::unique_ptr<Node>(y);}y->left = std::move(x->right);x->parent = y;}// 右旋操作void rightRotate(Node* y) {Node* x = y->left.get();y->left = std::move(x->right);if (x->right) {x->right->parent = y;}x->parent = y->parent;if (!y->parent) {root = std::unique_ptr<Node>(x);} else if (y == y->parent->right.get()) {y->parent->right = std::unique_ptr<Node>(x);} else {y->parent->left = std::unique_ptr<Node>(x);}x->right = std::move(y->left);y->parent = x;}// 插入修復void insertFixup(Node* z) {while (z->parent && z->parent->color == Color::RED) {if (z->parent == z->parent->parent->left.get()) {Node* y = z->parent->parent->right.get();if (y && y->color == Color::RED) {z->parent->color = Color::BLACK;y->color = Color::BLACK;z->parent->parent->color = Color::RED;z = z->parent->parent;} else {if (z == z->parent->right.get()) {z = z->parent;leftRotate(z);}z->parent->color = Color::BLACK;z->parent->parent->color = Color::RED;rightRotate(z->parent->parent);}} else {Node* y = z->parent->parent->left.get();if (y && y->color == Color::RED) {z->parent->color = Color::BLACK;y->color = Color::BLACK;z->parent->parent->color = Color::RED;z = z->parent->parent;} else {if (z == z->parent->left.get()) {z = z->parent;rightRotate(z);}z->parent->color = Color::BLACK;z->parent->parent->color = Color::RED;leftRotate(z->parent->parent);}}}root->color = Color::BLACK;}// 查找最小節點Node* minimum(Node* node) const {while (node->left) {node = node->left.get();}return node;}// 查找最大節點Node* maximum(Node* node) const {while (node->right) {node = node->right.get();}return node;}// 查找后繼節點Node* successor(Node* node) const {if (node->right) {return minimum(node->right.get());}Node* y = node->parent;while (y && node == y->right.get()) {node = y;y = y->parent;}return y;}// 查找前驅節點Node* predecessor(Node* node) const {if (node->left) {return maximum(node->left.get());}Node* y = node->parent;while (y && node == y->left.get()) {node = y;y = y->parent;}return y;}// 刪除修復void deleteFixup(Node* x, Node* parent) {while (x != root.get() && (x == nullptr || x->color == Color::BLACK)) {if (x == parent->left.get()) {Node* w = parent->right.get();if (w->color == Color::RED) {w->color = Color::BLACK;parent->color = Color::RED;leftRotate(parent);w = parent->right.get();}if ((w->left == nullptr || w->left->color == Color::BLACK) &&(w->right == nullptr || w->right->color == Color::BLACK)) {w->color = Color::RED;x = parent;parent = x->parent;} else {if (w->right == nullptr || w->right->color == Color::BLACK) {if (w->left) w->left->color = Color::BLACK;w->color = Color::RED;rightRotate(w);w = parent->right.get();}w->color = parent->color;parent->color = Color::BLACK;if (w->right) w->right->color = Color::BLACK;leftRotate(parent);x = root.get();parent = nullptr;}} else {Node* w = parent->left.get();if (w->color == Color::RED) {w->color = Color::BLACK;parent->color = Color::RED;rightRotate(parent);w = parent->left.get();}if ((w->right == nullptr || w->right->color == Color::BLACK) &&(w->left == nullptr || w->left->color == Color::BLACK)) {w->color = Color::RED;x = parent;parent = x->parent;} else {if (w->left == nullptr || w->left->color == Color::BLACK) {if (w->right) w->right->color = Color::BLACK;w->color = Color::RED;leftRotate(w);w = parent->left.get();}w->color = parent->color;parent->color = Color::BLACK;if (w->left) w->left->color = Color::BLACK;rightRotate(parent);x = root.get();parent = nullptr;}}}if (x) x->color = Color::BLACK;}// 中序遍歷輔助函數void inorderTraversal(Node* node) const {if (node) {inorderTraversal(node->left.get());std::cout << node->data << " ";inorderTraversal(node->right.get());}}// 查找節點輔助函數Node* findNode(const T& value) const {Node* current = root.get();while (current) {if (compare(value, current->data)) {current = current->left.get();} else if (compare(current->data, value)) {current = current->right.get();} else {return current;}}return nullptr;}public:RedBlackTree() : root(nullptr), compare(), treeSize(0) {}// 插入操作void insert(const T& value) {Node* z = new Node(value);Node* y = nullptr;Node* x = root.get();while (x) {y = x;if (compare(z->data, x->data)) {x = x->left.get();} else {x = x->right.get();}}z->parent = y;if (!y) {root = std::unique_ptr<Node>(z);} else if (compare(z->data, y->data)) {y->left = std::unique_ptr<Node>(z);} else {y->right = std::unique_ptr<Node>(z);}z->color = Color::RED;insertFixup(z);treeSize++;}// 刪除操作bool remove(const T& value) {Node* z = findNode(value);if (!z) return false;Node* y = z;Node* x;Color yOriginalColor = y->color;if (!z->left) {x = z->right.get();transplant(z, std::move(z->right));} else if (!z->right) {x = z->left.get();transplant(z, std::move(z->left));} else {y = minimum(z->right.get());yOriginalColor = y->color;x = y->right.get();if (y->parent == z) {if (x) x->parent = y;} else {transplant(y, std::move(y->right));y->right = std::move(z->right);y->right->parent = y;}transplant(z, std::move(std::unique_ptr<Node>(y)));y->left = std::move(z->left);y->left->parent = y;y->color = z->color;}if (yOriginalColor == Color::BLACK) {deleteFixup(x, y->parent);}treeSize--;return true;}// 查找操作bool contains(const T& value) const {return findNode(value) != nullptr;}// 獲取元素數量size_t size() const {return treeSize;}// 判斷是否為空bool empty() const {return treeSize == 0;}// 中序遍歷void inorder() const {inorderTraversal(root.get());std::cout << std::endl;}// 移植輔助函數void transplant(Node* u, std::unique_ptr<Node> v) {Node* uParent = u->parent;if (!uParent) {root = std::move(v);} else if (u == uParent->left.get()) {uParent->left = std::move(v);} else {uParent->right = std::move(v);}if (v) {v->parent = uParent;}}
};#endif // RED_BLACK_TREE_HPP
這段代碼實現了一個模板類 RedBlackTree,包含了紅黑樹的基本操作。代碼中使用了智能指針管理內存,確保內存安全。同時,通過插入修復和刪除修復函數來維護紅黑樹的性質。
紅黑樹的應用
紅黑樹在計算機科學中有廣泛的應用,主要包括:
- C++ 標準庫:STL 中的 map 和 set 就是基于紅黑樹實現的,保證了元素的有序性和高效的插入、刪除、查找操作。
- 操作系統:Linux 內核中的完全公平調度器(CFS)使用紅黑樹來管理進程調度。
- 數據庫系統:許多數據庫系統使用紅黑樹來索引數據,提高查詢效率。
- 其他應用:如 Java 的 TreeMap 和 TreeSet、Python 的 SortedContainers 等。
紅黑樹與其他數據結構的比較
紅黑樹與其他平衡二叉搜索樹(如 AVL 樹、B 樹等)相比,具有以下特點:
- 與 AVL 樹相比:紅黑樹的平衡性要求不如 AVL 樹嚴格,因此插入和刪除操作的旋轉次數更少,但查找操作的效率略低。
- 與 B 樹相比:紅黑樹是二叉樹,而 B 樹是多叉樹,更適合存儲在磁盤等外部存儲設備上。
- 與哈希表相比:紅黑樹可以保證元素的有序性,而哈希表不能,但哈希表的平均查找時間復雜度為 O (1),比紅黑樹更快。
總結
紅黑樹作為一種重要的數據結構,憑借其良好的平衡性和高效的操作性能,在計算機科學領域得到了廣泛應用。通過本文的介紹,讀者應該對紅黑樹的原理、實現和應用有了全面的了解。掌握紅黑樹不僅有助于理解各種高級算法和數據結構,也能在實際編程中發揮重要作用。
希望本文對您理解 C++ 紅黑樹有所幫助!
?
附:恩師博客hnjzsyjyj-CSDN博客