117.info
人生若只如初见

如何在C++中手动实现一个红黑树

要手动实现一个红黑树,在C++中,你需要首先定义红黑树的节点结构,并实现红黑树的插入、删除、查找等基本操作。下面是一个简单的红黑树的实现示例:

#include 

enum Color { RED, BLACK };

struct Node {
    int key;
    Color color;
    Node* left;
    Node* right;
    Node* parent;

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

class RedBlackTree {
private:
    Node* root;

    void rotateLeft(Node* node) {
        Node* rightChild = node->right;
        node->right = rightChild->left;
        if (rightChild->left != nullptr) {
            rightChild->left->parent = node;
        }
        rightChild->parent = node->parent;
        if (node->parent == nullptr) {
            root = rightChild;
        } else if (node == node->parent->left) {
            node->parent->left = rightChild;
        } else {
            node->parent->right = rightChild;
        }
        rightChild->left = node;
        node->parent = rightChild;
    }

    void rotateRight(Node* node) {
        Node* leftChild = node->left;
        node->left = leftChild->right;
        if (leftChild->right != nullptr) {
            leftChild->right->parent = node;
        }
        leftChild->parent = node->parent;
        if (node->parent == nullptr) {
            root = leftChild;
        } else if (node == node->parent->right) {
            node->parent->right = leftChild;
        } else {
            node->parent->left = leftChild;
        }
        leftChild->right = node;
        node->parent = leftChild;
    }

    void fixInsertion(Node* node) {
        while (node != root && node->parent->color == RED) {
            if (node->parent == node->parent->parent->left) {
                Node* uncle = node->parent->parent->right;
                if (uncle != nullptr && uncle->color == RED) {
                    node->parent->color = BLACK;
                    uncle->color = BLACK;
                    node->parent->parent->color = RED;
                    node = node->parent->parent;
                } else {
                    if (node == node->parent->right) {
                        node = node->parent;
                        rotateLeft(node);
                    }
                    node->parent->color = BLACK;
                    node->parent->parent->color = RED;
                    rotateRight(node->parent->parent);
                }
            } else {
                Node* uncle = node->parent->parent->left;
                if (uncle != nullptr && uncle->color == RED) {
                    node->parent->color = BLACK;
                    uncle->color = BLACK;
                    node->parent->parent->color = RED;
                    node = node->parent->parent;
                } else {
                    if (node == node->parent->left) {
                        node = node->parent;
                        rotateRight(node);
                    }
                    node->parent->color = BLACK;
                    node->parent->parent->color = RED;
                    rotateLeft(node->parent->parent);
                }
            }
        }
        root->color = BLACK;
    }

    void insert(Node* node) {
        Node* current = root;
        Node* parent = nullptr;
        while (current != nullptr) {
            parent = current;
            if (node->key < current->key) {
                current = current->left;
            } else {
                current = current->right;
            }
        }
        node->parent = parent;
        if (parent == nullptr) {
            root = node;
        } else if (node->key < parent->key) {
            parent->left = node;
        } else {
            parent->right = node;
        }
        node->color = RED;
        fixInsertion(node);
    }

public:
    RedBlackTree() : root(nullptr) {}

    void insert(int key) {
        Node* node = new Node(key);
        insert(node);
    }

    void printInOrderTraversal(Node* node) {
        if (node == nullptr) {
            return;
        }
        printInOrderTraversal(node->left);
        std::cout << node->key << " ";
        printInOrderTraversal(node->right);
    }

    void print() {
        printInOrderTraversal(root);
        std::cout << std::endl;
    }
};

int main() {
    RedBlackTree tree;

    tree.insert(10);
    tree.insert(20);
    tree.insert(30);
    tree.insert(15);
    tree.insert(25);

    tree.print();

    return

未经允许不得转载 » 本文链接:https://www.117.info/ask/fef60AzsIBgVfAFA.html

推荐文章

  • C#装箱和拆箱是什么意思

    装箱和拆箱是指将值类型转换为引用类型和将引用类型转换为值类型的过程。
    装箱:将值类型数据转换为引用类型数据的过程称为装箱。在装箱时,会将值类型数据...

  • C#解释反射的用途是什么

    反射是在运行时动态获取程序元数据和操作程序对象的技术。在C#中,反射可以用来检查和操作程序集、类型、成员等,可以实现以下功能: 动态创建对象:通过反射可以...

  • 在C#中怎么实现多线程

    在C#中实现多线程可以使用Thread类或Task类。 使用Thread类: using System;
    using System.Threading; class Program
    { static void Main() { Thread ...

  • C#全局使用指令有哪些优势

    C#全局使用指令有以下几个优势: 代码重用性:全局使用指令可以在多个文件中引用相同的命名空间或类型,从而避免重复编写代码,提高代码复用性。 代码可读性:通...

  • C++标准模板库与红黑树的关系

    C++标准模板库(STL)是C++语言的标准库之一,其中包含了许多常用的数据结构和算法。STL中的容器类和算法类大部分都是基于模板的,可以通过模板参数来指定具体的...

  • 深入理解红黑树中的删除过程及其复杂性

    红黑树是一种自平衡的二叉搜索树,其删除过程相对于添加和查找操作来说更为复杂。删除节点时需要考虑多种情况,包括删除节点的子节点情况、兄弟节点的颜色以及路...

  • 探索红黑树的插入操作及其调整策略

    红黑树是一种自平衡的二叉查找树,其插入操作需要经过一系列的调整来保持树的平衡性质。以下是红黑树的插入操作及其调整策略: 将新节点插入到红黑树中的合适位置...

  • 在C++中实现红黑树的基本结构

    红黑树是一种自平衡的二叉搜索树,其基本结构可以通过以下代码实现:
    #include using namespace std; enum Color {RED, BLACK}; struct Node { int data; C...