【C++学习】红黑树

最新推荐文章于 2024-12-25 21:53:58 发布

Cassooo_

最新推荐文章于 2024-12-25 21:53:58 发布

阅读量973

点赞数 23

CC 4.0 BY-SA版权

分类专栏： C++ 文章标签：学习 c++ 红黑树

本文链接：https://blog.csdn.net/weixin_44906102/article/details/138802299

5.2 检测新节点插入后，红黑树的性质是否造到破坏

5.2.1 情况一：cur为红，p为红，g为黑，u存在且为红

5.2.2 情况二：cur为红，p为红，g为黑，u不存在/u存在且为黑

六、红黑树的验证

七、红黑树模拟实现STL中的map与set

一、红黑树的概念

红黑树，是一种 二叉搜索树 ，但 在每个结点上增加一个存储位表示结点的颜色，可以是 Red或B lack 。通过对 任何一条从根到叶子的路径上各个结点着色方式的限制，红黑树确保没有一条路径 会比其他路径长出俩倍 ，因而是 接近平衡 的。

二、红黑树的性质

每个结点不是红色就是黑色
根节点是黑色的
如果一个节点是红色的，则它的两个孩子结点是黑色的
对于每个结点，从该结点到其所有后代叶结点的简单路径上，均包含相同数目的黑色结点
每个叶子结点都是黑色的(此处的叶子结点指的是空结点)

三、红黑树节点的定义

// 节点的颜色
enum Color{RED, BLACK};

// 红黑树节点的定义
template<class ValueType>
struct RBTreeNode
{
    RBTreeNode(const ValueType& data = ValueType()，Color color = RED)
        : _pLeft(nullptr), _pRight(nullptr), _pParent(nullptr)
        , _data(data), _color(color)
    {}

    RBTreeNode<ValueType>* _pLeft;   // 节点的左孩子
    RBTreeNode<ValueType>* _pRight;  // 节点的右孩子
    RBTreeNode<ValueType>* _pParent; // 节点的双亲(红黑树需要旋转，为了实现简单给出该字段)
    ValueType _data;            // 节点的值域
    Color _color;               // 节点的颜色
};

四、红黑树的结构

为了后续实现关联式容器简单，红黑树的实现中增加一个头结点，因为跟节点必须为黑色，为了与根节点进行区分，将头结点给成黑色，并且让头结点的 pParent 域指向红黑树的根节点， pLeft域指向红黑树中最小的节点，_pRight 域指向红黑树中最大的节点，如下：

五、红黑树的插入操作

红黑树是在二叉搜索树的基础上加上其平衡限制条件，因此红黑树的插入可分为两步：

5.1 按照二叉搜索的树规则插入新节点

template<class ValueType>
class RBTree
{
    //……
    bool Insert(const ValueType& data)
    {
        PNode& pRoot = GetRoot();
        if (nullptr == pRoot)
        {
            pRoot = new Node(data, BLACK);

            // 根的双亲为头节点
            pRoot->_pParent = _pHead;
            _pHead->_pParent = pRoot;
        }
        else
        {
             // 1. 按照二叉搜索的树方式插入新节点
             // 2. 检测新节点插入后，红黑树的性质是否造到破坏，
             //   若满足直接退出，否则对红黑树进行旋转着色处理
        }

        // 根节点的颜色可能被修改，将其改回黑色
        pRoot->_color = BLACK;
        _pHead->_pLeft = LeftMost();
        _pHead->_pRight = RightMost();
        return true;
    }

private:
    PNode& GetRoot(){ return _pHead->_pParent; }

    // 获取红黑树中最小节点，即最左侧节点
    PNode LeftMost();

    // 获取红黑树中最大节点，即最右侧节点
    PNode RightMost();

private:
    PNode _pHead;
};