赞
踩
从概念上来讲,它是一种二叉搜索树,但每一个节点不是红色就是黑色,它保证最长路径不会比其他路径长出两倍
为什么能保证,就要看下面几条性质:
1. 每个结点不是红色就是黑色
2. 根节点是黑色的
3. 如果一个节点是红色的,则它的两个孩子结点是黑色的
4. 对于每个结点,从该结点到其所有后代叶结点的简单路径上,均 包含相同数目的黑色结点
5. 每个叶子结点都是黑色的(此处的叶子结点指的是空结点)
为什么满足了上面的性质就可以保证最长的不超过最短的两倍?
因为最短路径肯定是全黑,但是最长路径是黑红黑红,因为其保证了每条路径黑色节点数量要一致,所以最长的黑红最多也就等于两倍的黑
这里会跟AVL树有所不同,因为这里之后要复用到map,所以会稍作修改
首先为了区分颜色,我们就可以用到枚举:
- enum Color
- {
- Red,
- Black
- };
然后创建节点的结构体,这里跟AVLTree差不多
- template<class K,class V>
- struct RBTreeNode
- {
- pair<K, V> _kv;
- RBTreeNode<K, V>* _parent;
- RBTreeNode<K, V>* _left;
- RBTreeNode<K, V>* _right;
-
- Color _color;
-
- RBTreeNode(const pair<K,V>& kv)
- :_kv(kv)
- , _parent(nullptr)
- , _left(nullptr)
- , _right(nullptr)
- , _color(Red)
- {}
-
- };
顺便写一个构造函数,也很简单,把指针置为空,将里面的值用外面的拷贝一份,这里要注意的是,为什么插入节点的颜色要是红色。
这里我们插入节点是一定要给颜色的,给颜色就会出现两种情况:
3. 如果一个节点是红色的,则它的两个孩子结点是黑色的
4. 对于每个结点,从该结点到其所有后代叶结点的简单路径上,均 包含相同数目的黑色结点
如果我们插入红色,有可能会违反3,因为它的父亲节点可能是红色的,但如果我们插入黑色节点,那就一定会违反4,因为你在一条支路里面插入黑色节点,如何确保其他路上的黑色节点数量一样就是个问题
先简单写一个红黑树的大框架出来
- template<class K, class V>
- class RBTree
- {
- typedef RBTreeNode<K, V> Node;
-
-
-
- private:
- Node* _root = nullptr;
- };
然后就可以考虑写插入了,这里需要提醒一下,是否还记得AVL树的旋转,这里红黑树也需要用到,如果不记得,这里放上另一篇博客的链接可供参考:
ok,最基本的先写上,如果这是一个空树,那最开始插入的就是根节点:
-
- //判断这是否为空树,是就直接插入
- if (_root == nullptr)
- {
- _root = new Node(kv);
- //根节点一定要是黑色
- _root->_color = Black;
- return true;
- }
然后,就可以开始插入了,插入部分根AVL基本一致
- Node* cur = _root;
- Node* parent = cur->_parent;
- while (cur)
- {
- if (cur->_kv.first > kv.first)
- {
- parent = cur;
- cur = cur->_left;
- }
- else if (cur->_kv.first < kv.first)
- {
- parent = cur;
- cur = cur->_right;
- }
- else
- return false;
-
- cur = new Node(kv);
- cur->_color = Red;
- if (cur = parent->_left)
- parent->_left = cur;
- else
- parent->_right = cur;
然后就是我们需要处理的地方,我们就开始判断:
假设,我们在这个位置插入,肯定是没问题的
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。