红黑树算法及代码(C++版)

红黑树的定义

1.节点非黑即红。
2.根节点为黑。
3.红色节点不连续。(插入时,做判定)
4.完美黑色平衡:根节点到NULL节点,经过黑色节点数目相同。

数据结构

节点结构

struct RedBlackNode{
    int _key;
    int _value;
    RedBlackNode *_left = NULL;
    RedBlackNode *_right = NULL;
    RedBlackNode *_parent = NULL;
    Color color = RED;
};

建立这棵树的规则

判定不符合性质3

满足while则不符合性质3

while(parent && parent->color == RED)

1.变色(uncle红,parent红)

parent、uncle-->黑
grandParent-->红
下一个可能出现不符合规则的,为grandParent节点,下次判定cur=grandParent,parent = grandParent->parent

PS:通过变色,当出现parent为根节点时,满足性质3,退出判定。

2.旋转(uncle黑,parent红)

parent-->黑
grandParent-->红 (经过旋转后,parent成为grandparent,grandparent成为parent)

类型 旋转情况
LL 右旋
LR 左旋后右旋
RR 左旋
RL 右旋后左旋

PS:
1.LR的本质是转换成LL。所以变色在左旋后。(RL同理)
2.旋转必结束判定

检查这棵树是否满足要求

检查条件:红黑树每条路径上黑色节点数量相同

外封装:

bool RBTree::Check(){
  int blackNum = 0;
  RBTreeNode *cur = _root;
  while(cur){
    if(cur->_color == BLACK)
      blackNum++;
    cur = cur->_left;//任意方向
  }
  int CBNum = 0;//count Black number
  return _Check(_root, blackNum, CBNum);
}

检查黑色节点是否满足要求

bool RBTree::_Check(RBTreeNode *root, int blackNum, int CBNum)
{
  if(root == NULL)
    return true;
  if(root->_color == BLACK)
  {
    CBNum++;
    if(root->_left == NULL && root->_right == NULL){
      return true;
    }
    else{
      cout << "叶子节点为" << root->_key << "路径的黑色节点数目与最左侧支路的黑色节点数目不相等!" << endl;
      return false;
    }
  }else if(root->_parent && root->_parent->_color == RED){
    //判断是否存在连续的两个红色节点
    cout << root->_parent->_key << " 和 "  << root->_key << "为两个连续的红色节点" << endl;
    return false;
  }
  return _Check(root->_left, blackNum, CBNum) && _Check(root->_right, blackNum, CBNum);
}
#include <iostream>
#include <cmath>
#include <vector>
#include <random>

using namespace std;

/*请在这里输入这个程序的功能或者作用*/
enum Color{RED, BLACK};
struct RedBlackNode{
    int _key;
    int _value;
    RedBlackNode *_left = NULL;
    RedBlackNode *_right = NULL;
    RedBlackNode *_parent = NULL;
    Color color = RED;
};


class RedBlackTree{
public:
    RedBlackTree():
        _root(NULL)
    {}
    void RotateL(RedBlackNode *&node){
        RedBlackNode *parent = node->_parent;
        RedBlackNode *subR = node->_right;
        RedBlackNode *subRL = subR->_left;

        //需要变动的是 node, subR, subRL, parent
        node->_parent = subR;
        node->_right = subRL;

        subR->_parent = parent;
        subR->_left = node;

        if(subRL)subRL->_parent = node;

        if(parent==NULL){
            _root = subR;
        }
        else{
            if(parent->_left == node){
                parent->_left = subR;
            }else if(parent->_right == node){
                parent->_right = subR;
            }
        }
        //根节点换人
        node = subR;
    }
    void RotateR(RedBlackNode *&node){
        RedBlackNode *parent = node->_parent;
        RedBlackNode *subL = node->_left;
        RedBlackNode *subLR = subL->_right;
        
        //node, subL, subLR, parent
        node->_left = subLR;
        node->_parent = subL;

        subL->_parent = parent;
        subL->_right = node;

        if(subLR)subLR->_parent = node;

        if(parent == NULL){
            _root = subL;
        }else{
            if(parent->_left == node){
                parent->_left = subL;
            }else if(parent->_right == node){
                parent->_right = subL;
            }
        }
        node = subL;
    }
    RedBlackNode* CreateNode(int key, int value){
        RedBlackNode *temp = new RedBlackNode;
        temp->_key = key; temp->_value = value;
        return temp;
    }
    bool Insert(int key, int value){
        if(_root == NULL){
            _root = CreateNode(key, value);
            _root->color = BLACK;
            return true;
        }
        RedBlackNode *cur = _root, *parent = NULL;
        while(cur){
            parent = cur;
            if(key < cur->_key){
                cur = cur->_left;
            }else if(key > cur->_key){
                cur = cur->_right;
            }else return false;
        }
        cur = CreateNode(key, value);
        if(key < parent->_key){
            parent->_left = cur;
        }else if(key > parent->_key){
            parent->_right = cur;
        }
        cur->_parent = parent;

        while(parent && parent->color == RED){
            RedBlackNode* grandParent = parent->_parent;
            if(parent == grandParent->_left){
                RedBlackNode *uncle = grandParent->_right;
                if(uncle && uncle->color == RED){
                    grandParent->color = RED;
                    parent->color = BLACK;
                    uncle->color = BLACK;
                    
                    cur = grandParent;
                    parent = cur->_parent;
                }else if((uncle && uncle->color == BLACK) || uncle == NULL){
                    if(cur == parent->_left){
                        parent->color = BLACK;
                        grandParent->color = RED;
                        RotateR(grandParent);
                    }else if(cur == parent->_right){
                        RotateL(parent);
                        parent->color = BLACK;
                        grandParent->color = RED;
                        RotateR(grandParent);
                    }
                    break;
                }
            }else if(parent == grandParent->_right){
                RedBlackNode *uncle = grandParent->_left;
                if(uncle && uncle->color == RED){
                    grandParent->color = RED;
                    parent->color = BLACK;
                    uncle->color = BLACK;
                    
                    cur = grandParent;
                    parent = cur->_parent;
                }else if((uncle && uncle->color == BLACK) || uncle == NULL){
                    if(cur == parent->_right){
                        parent->color = BLACK;
                        grandParent->color = RED;
                        RotateL(grandParent);
                    }else if(cur == parent->_left){
                        RotateR(parent);
                        parent->color = BLACK;
                        grandParent->color = RED;
                        RotateL(grandParent);
                    }
                    break;
                }
            }
        }
        _root->color = BLACK;
        return true;        
    }
    void InOrder(){
        if(_root == NULL){
            cout << "No";
            return;
        }
        _InOrder(_root);
    }
    void _InOrder(RedBlackNode *a){
        if(a==NULL)return;
        _InOrder(a->_left);
        string str_color = a->color==RED ? "RED": "BLACK";
        cout << "key:"<< a->_key << "value:" << a->_value << " Color:" << str_color;
        if(a->_parent) cout << "parent_key:" << a->_parent->_key;
        cout << endl;
        _InOrder(a->_right);
        return;
    }
    bool Check(){
        int blackNum = 0;
        RedBlackNode *cur = _root;
        while(cur){
            if(cur->color == BLACK)
            blackNum++;
            cur = cur->_left;//任意方向
        }
        int CBNum = 0;//count Black number
        return _Check(_root, blackNum, CBNum);
    }
    bool _Check(RedBlackNode *root, int blackNum, int CBNum)
    {
        if(root == NULL)
            return true;
        if(root->color == BLACK)
        {
            CBNum++;
            if(root->_left == NULL && root->_right == NULL){
                if(blackNum == CBNum)
                    return true;
                else{
                    cout << "叶子节点为" << root->_key << "路径的黑色节点数目与最左侧支路的黑色节点数目不相等!" << endl;
                    return false;
                }
            }
        }else if(root->_parent && root->_parent->color == RED){
            //判断是否存在连续的两个红色节点
            cout << root->_parent->_key << " 和 "  << root->_key << "为两个连续的红色节点" << endl;
            return false;
        }
        return _Check(root->_left, blackNum, CBNum) && _Check(root->_right, blackNum, CBNum);
    }
private:
    RedBlackNode* _root;
};

int main()
{
    RedBlackTree tree;
    vector<int> nums = {16,3,7,11,9,26,18,14,15,20,100,0,1,2,5};
    for(int i=0; i<nums.size(); i++){
        tree.Insert(nums[i], i);
    }
    tree.InOrder();
    cout << "check:" << tree.Check();
    return 0;
}


参考资料
红黑树知识:脑子里(忘记哪里学的了)
红黑树代码参考(该代码使用了模板,我用了int形式自己写了一遍)

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,922评论 6 497
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,591评论 3 389
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 160,546评论 0 350
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,467评论 1 288
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,553评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,580评论 1 293
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,588评论 3 414
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,334评论 0 270
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,780评论 1 307
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,092评论 2 330
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,270评论 1 344
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,925评论 5 338
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,573评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,194评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,437评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,154评论 2 366
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,127评论 2 352