码迷,mamicode.com
首页 > 其他好文 > 详细

平衡搜索树(二) Rb 红黑树

时间:2016-04-28 00:18:18      阅读:248      评论:0      收藏:0      [点我收藏+]

标签:

Rb树简介

红黑树是一棵二叉搜索树,它在每个节点上增加了一个存储位来表示节点的颜色,可以是Red或Black。通过对任何一条从根到叶子简单 路径上的颜色来约束,红黑树保证最长路径不超过最短路径的两倍,因而近似于平衡。(性质3、性质4保证了红黑树最长的路径不超过最短路径的两倍

如图所示:

技术分享


红黑树是满足下面红黑性质的二叉搜索树


1. 每个节点,不是红色就是黑色的

2. 根节点是黑色的

3. 如果一个节点是红色的,则它的两个子节点是黑色的

4. 对每个节点,从该节点到其所有后代叶节点的简单路径上,均包含相同数目的黑色节点。

 

好,现在了解了红黑树的特性之后,我们来开始创建红黑树。

 

//红黑树的节点的结构
 1 template<class K,class V>         
 2 struct RedBlaceTreeNode
 3 {
 4     typedef RedBlaceTreeNode<K, V> Node;
 5     RedBlaceTreeNode(const K& key,const V& value)
 6         :_left(NULL), _right(NULL), _parent(NULL)
 7         , _key(key), _value(value), _col(RED)
 8     {}
 9 
10     Node* _left;
11     Node* _right;
12     Node* _parent;
13     K _key;
14     V _value;
15     Sign _col;
16 };

 

//红黑树的结构

template<class K,class V>
class RedBlaceTree
{
    typedef RedBlaceTreeNode<K, V> Node;
public:
    RedBlaceTree()
        :_root(NULL)
    {}
public:
    bool _Push(const K& key, const V& value);
    void _LeftSpin(Node*& parent);
    void _RightSpin(Node*& parent);
    void _LeftRightSpin(Node*& parent);
    void _RightLeftSpin(Node*& parent);

    /*Check*/
    //1. 每个节点,不是红色就是黑色的
    //2. 根节点是黑色的
    //3. 如果一个节点是红色的,则它的两个子节点是黑色的
    //4. 对每个节点,从该节点到其所有后代叶节点的简单路径上,均包含相同数目的黑色节点
    void _Inorder(){ Inorder(_root); }
    void Inorder(Node* root);
protected:
    Node* _root;
};

 

 

//创建红黑树的函数 _Push()

插入的几种情况

cur为当前节点,p为父节点,g为祖父节点,u为叔叔节点

1.第一种情况 cur为红,p为红,g为黑,u存在且为红 则将p,u改为黑,g改为红,然后把g当成cur,继续向上调整。( 还挺简单  O(∩_∩)O   继续往下看↓↓)

技术分享

 

2.第二种情况 cur为红,p为红,g为黑,u不存在/u为黑 p为g的左孩子,cur为p的左孩子,则进行右单旋转;相反,p为g的右孩子,cur为p的右孩子,则进行左单旋转 p、g变色--p变黑,g变红

技术分享

3.第三种情况 cur为红,p为红,g为黑,u不存在/u为黑
p为g的左孩子,cur为p的右孩子,则针对p做左单旋转;相反,p为g的右孩子,cur为p的左孩子,则针对p做右单旋转 则转换成了情况2

 

技术分享

 

 

template<class K, class V>
bool RedBlaceTree<K, V>::_Push(const K& key, const V& value)
{
    Node* cur = _root;
    Node* parent = NULL;
    if (_root == NULL)
    {
        _root = new Node(key, value);
        _root->_col = BLACK;
    }
    else
    {
        while (cur)
        {
            parent = cur;
            if (key > cur->_key)
            {
                cur = cur->_right;
            }
            else if (key < cur->_key)
            {
                cur = cur->_left;
            }
            else
            {
                return false;
            }
        }
        cur = new Node(key, value);
        if (key > parent->_key)
        {
            parent->_right = cur;
            cur->_parent = parent;
        }
        else
        {
            parent->_left = cur;
            cur->_parent = parent;
        }

        if (parent->_col == RED)     //如果父亲节点为红色就需要调整
        {
            bool sign = false;      //标记位,用来标记grandparent是否为根节点
            while (cur->_parent)
            {
                Node* grandparent = NULL;
                Node* uncle = NULL;
                parent = cur->_parent;
                grandparent = parent->_parent;

                if (grandparent)       //如果祖父节点存在
                {
                    if (grandparent == _root)
                    {
                        sign = true;    //标记祖父节点是否为根节点
                    }

                    if (parent->_key > grandparent->_key)   //确定叔父节点
                    {
                        uncle = grandparent->_left;
                    }
                    else
                    {
                        uncle = grandparent->_right;
                    }


                    //第一种情况:cur为红,p为红,g为黑,u存在且为红 
                    if (uncle && uncle->_col == RED)
                    {
                        parent->_col = BLACK;
                        uncle->_col = BLACK;
                        grandparent->_col = RED;
                        if (sign)
                        {
                            break;
                        }
                        cur = grandparent;
                    }

                    //第二种情况:cur为红,p为红,g为黑,u不存在/u为黑(单旋)
                    //第三种情况:cur为红,p为红,g为黑,u不存在/u为黑(双旋)
                    else
                    {
                        //第二
                        if (grandparent->_left == parent && parent->_left == cur)
                        {
                            grandparent->_col = RED;
                            parent->_col = BLACK;
                            _RightSpin(grandparent);
                            if (sign)
                            {
                                _root = grandparent;
                            }
                            break;
                        }
                        else if (grandparent->_right == parent && parent->_right == cur)
                        {
                            grandparent->_col = RED;
                            parent->_col = BLACK;
                            _LeftSpin(grandparent);
                            if (sign)
                            {
                                _root = grandparent;
                            }
                            break;
                        }
                        //第三
                        else if (grandparent->_left == parent && parent->_right == cur)
                        {
                            grandparent->_col = RED;
                            cur->_col = BLACK;
                            _LeftRightSpin(grandparent);
                            if (sign)
                            {
                                _root = grandparent;
                            }
                            break;
                        }
                        else if (grandparent->_right == parent && parent->_left == cur)
                        {
                            grandparent->_col = RED;
                            cur->_col = BLACK;
                            _RightLeftSpin(grandparent);
                            if (sign)
                            {
                                _root = grandparent;
                            }
                            break;
                        }
                    }
                }
                else   //cur 的上面只有一层
                {
                    break;
                }
            }
        }
    }
    _root->_col = BLACK;
}

 

平衡搜索树(二) Rb 红黑树

标签:

原文地址:http://www.cnblogs.com/shihaokiss/p/5440999.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!