c++怎么实现一个红黑树_c++平衡二叉搜索树插入与旋转算法

红黑树通过着色规则和旋转保持平衡,插入后修复以确保根黑、无连续红、黑高一致,C++实现包含左旋右旋与insertFixup,最终中序遍历验证有序性。

c++怎么实现一个红黑树_c++平衡二叉搜索树插入与旋转算法

红黑树是一种自平衡的二叉搜索树(BST),它通过为每个节点着色(红色或黑色)并遵循特定规则来保持树的近似平衡,从而保证查找、插入和删除操作的时间复杂度为 O(log n)。下面用 C++ 实现一个基础的红黑树,包含插入操作和必要的旋转调整逻辑。

红黑树的性质

在实现前先明确红黑树必须满足的五条性质:

每个节点是红色或黑色根节点是黑色所有叶子(NULL 节点)视为黑色如果一个节点是红色,则它的两个子节点都是黑色(不能有两个连续的红色节点)从任一节点到其每个叶子的所有路径都包含相同数目的黑色节点(黑高一致)

节点结构定义

每个节点需要存储值、颜色、左右子节点指针和父节点指针:

enum Color { RED, BLACK };

struct Node {int data;Color color;Node left, right, *parent;

Node(int data) : data(data), color(RED), left(nullptr), right(nullptr), parent(nullptr) {}

};

立即学习“C++免费学习笔记(深入)”;

左旋与右旋操作

旋转是维持红黑树平衡的核心操作。左旋用于处理右倾情况,右旋用于处理左倾中的特定问题。

void leftRotate(Node* &root, Node* x) {    Node* y = x->right;    x->right = y->left;    if (y->left != nullptr)        y->left->parent = x;    y->parent = x->parent;
if (x->parent == nullptr)    root = y;else if (x == x->parent->left)    x->parent->left = y;else    x->parent->right = y;y->left = x;x->parent = y;

}

void rightRotate(Node &root, Node y) {Node* x = y->left;y->left = x->right;if (x->right != nullptr)x->right->parent = y;x->parent = y->parent;

if (y->parent == nullptr)    root = x;else if (y == y->parent->left)    y->parent->left = x;else    y->parent->right = x;x->right = y;y->parent = x;

}

插入与修复操作

插入新节点后,可能破坏红黑性质,需进行修复。新节点默认为红色,然后根据父节点颜色和叔节点状态分情况处理。

void insertFixup(Node* &root, Node* z) {    while (z != root && z->parent->color == RED) {        if (z->parent == z->parent->parent->left) {            Node* uncle = z->parent->parent->right;            if (uncle != nullptr && uncle->color == RED) {                // 情况1:叔节点为红                z->parent->color = BLACK;                uncle->color = BLACK;                z->parent->parent->color = RED;                z = z->parent->parent;            } else {                // 情况2:叔节点为黑且当前节点为右孩子                if (z == z->parent->right) {                    z = z->parent;                    leftRotate(root, z);                }                // 情况3:叔节点为黑且当前节点为左孩子                z->parent->color = BLACK;                z->parent->parent->color = RED;                rightRotate(root, z->parent->parent);            }        } else {            Node* uncle = z->parent->parent->left;            if (uncle != nullptr && uncle->color == RED) {                z->parent->color = BLACK;                uncle->color = BLACK;                z->parent->parent->color = RED;                z = z->parent->parent;            } else {                if (z == z->parent->left) {                    z = z->parent;                    rightRotate(root, z);                }                z->parent->color = BLACK;                z->parent->parent->color = RED;                leftRotate(root, z->parent->parent);            }        }    }    root->color = BLACK; // 根节点始终为黑}

void insert(Node &root, int data) {Node z = new Node(data);Node y = nullptr;Node x = root;

while (x != nullptr) {    y = x;    if (z->data data)        x = x->left;    else        x = x->right;}z->parent = y;if (y == nullptr)    root = z;else if (z->data data)    y->left = z;else    y->right = z;insertFixup(root, z);

}

完整使用示例

以下是一个简单的测试主函数:

#include using namespace std;

// 上述所有代码放在这里

void inorder(Node* root) {if (root != nullptr) {inorder(root->left);cout <data <right);}}

int main() {Node* root = nullptr;insert(root, 10);insert(root, 20);insert(root, 30);insert(root, 15);insert(root, 25);

cout << "Inorder traversal: ";inorder(root);cout << endl;return 0;

}

基本上就这些。这个实现涵盖了红黑树插入和旋转的核心机制。虽然没有包含删除操作(更复杂),但已足够理解其平衡原理。实际工程中可考虑使用 std::setstd::map,它们底层正是基于红黑树实现的。

以上就是c++++怎么实现一个红黑树_c++平衡二叉搜索树插入与旋转算法的详细内容,更多请关注创想鸟其它相关文章!

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1483403.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
c++怎么使用C++17的结构化绑定_c++多返回值与解构赋值语法详解
上一篇 2025年12月19日 07:12:41
C++怎么使用SFML进行2D游戏开发_C++游戏开发与SFML应用
下一篇 2025年12月19日 07:12:46

相关推荐

发表回复

登录后才能评论
关注微信