红黑树实现

1.红黑树的概念

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

2.红黑树的特性

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

3.红黑树插入实现

前面插入和搜索二叉树和平衡二叉树没有区别一样的访问方式，区别就在，他的节点的颜色
有三种情况
第一种：叔叔节点存在且为红色，那就让父亲节点和叔叔节点变黑（因为要求不能连续的红上面特性有写），再让祖父节点变红，为什么要变红，因为他可能是局部节点

第二种：叔叔节点不存在或者为黑，那就进行右旋，不过也有可能是左旋，看那边的节点比较高

第三种：这种和第二种差不多，区别就是cur等于父亲节点的right，那就是个折线，那就要进行双旋，双旋当然，也要和第二种分情况，看是先左旋还是先右旋

4.全部代码包含测试用例

#pragma once
#include <assert.h>
#include <vector>
#include <queue>
#include <time.h>

enum Colour
{
   
	RED,
	BLACK,
};

template<class K, class V>
struct RBTreeNode
{
   
	RBTreeNode<K, V>* _left;
	RBTreeNode<K, V>* _right;
	RBTreeNode<K, V>* _parent;
	pair<K, V> _kv;

	Colour _col;

	RBTreeNode(const pair<K, V>& kv)
		:_kv(kv)
		, _left(nullptr)
		, _right(nullptr)
		, _parent(nullptr)
		, _col(RED)
	{
   }
};

template<class K, class V>
struct RBTree
{
   
	typedef RBTreeNode<K, V> Node;
public:


	bool Insert(const pair<K, V>& kv)
	{
   
		// 1、搜索树的规则插入
		// 2、看是否违反平衡规则，如果违反就需要处理：旋转
		if (_root == nullptr)
		{
   
			_root = new Node(kv);
			_root->_col = BLACK;
			return true;
		}

		Node* parent = nullptr;
		Node* cur = _root;
		while (cur)
		{
   
			if (cur->_kv.first < kv.first)
			{
   
				parent = cur;
				cur = cur->_right;
			}
			else if (cur->_kv.first > kv.first)
			{
   
				parent = cur;
				cur = cur->_left;
			}
			else
			{
   
				return false;
			}
		}

		cur = new Node(kv);
		cur->_col = RED;
		if (parent->_kv.first < kv.first)
		{
   
			parent->_right = cur;
		}
		else
		{
   
			parent->_left = cur;
		}

		cur->_parent = parent;

		//存在连续的红色节点
		while (parent && parent->_col == RED)
		{
   
			Node* grandfater = parent->_parent;
			assert(grandfater);

			if (grandfater->_left == parent)
			{
   
				Node* uncle = grandfater->_right;
				// 情况一：
				if (uncle && uncle->_col == RED) // 叔叔存在且为红
				{
   
					// 变色
					parent->_col = uncle->_col = BLACK;
					grandfater->_col = RED;

					// 继续往上处理
					cur = grandfater;
					parent = cur->_parent;
				}
				else // 情况二：叔叔不存在 或者 叔叔存在且为黑
				{
    
					if (cur == parent->_left)
					{
   
						//     g
			            //   p
			            // c
						RotateR(grandfater);//右旋
						parent->_col = BLACK;
						grandfater->_col = RED;
					}
					else//情况三： cur等于parent的right形成折现    双旋 
					{
   
						//     g
					    //   p
					    //     c 
						RotateL(parent);
						RotateR(grandfater);
						cur->_col = BLACK;
						grandfater->_col = RED;
					}

					break;
				}
			}
			else//(grandfater->_right == parent)
			{
   
				Node* uncle = grandfater->_left;
				// 情况一：
				if (uncle && uncle->_col == RED)
				{
   
					// 变色
					parent->_col = uncle->_col = BLACK;
					grandfater->_col = RED;

					// 继续往上处理
					cur = grandfater;
					parent = cur->_parent;
				}
				else// 情况二：叔叔不存在 或者 叔叔存在且为黑
				{
   
					if (cur == parent->_right)
					{
   
						// g
                        //   p
                        //     c 
						RotateL(grandfater);//左旋

						parent->_col = BLACK;
						grandfater->_col = RED;
					}
					else//情况三： cur等于parent的right形成折现    双旋 
					{
   
						// g
				        //   p
				        // c
						RotateR(parent);
						RotateL(grandfater);
						cur->_col = BLACK;
						grandfater->_col = RED;
					}
					break;
				}
			}
		}

		_root->_col = BLACK;

		return true;
	}


	vector<vector<int>> levelOrder() {
   
		vector<vector<int>> vv;
		if (_root == nullptr)
			return vv;

		queue<Node*> q;
		int levelSize = 1;
		q.push(_root);

		while (!q.empty())
		{
   
			// levelSize控制一层一层出
			vector<int> levelV;
			while (levelSize--)
			{
   
				Node* front = q.front();
				q.pop();
				levelV.push_back(front->_kv.first);
				if (front->_left)
					q.push(front->_left);

				if (front->_right)
					q.push(front->_right);
			}
			vv.push_back(levelV);
			for (auto e : levelV)
			{
   
				cout << e << " ";
			}
			cout << endl;

			// 上一层出完，下一层就都进队列
			levelSize = q.size();
		}

		return vv;
	}
	void RotateL(Node* parent)
	{
   
		Node* subR = parent->_right;
		Node* subRL = subR->_left;

		parent->_right = subRL;
		if (subRL)
			subRL->_parent = parent;

		Node* ppNode = parent->_parent;

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

		if (parent == _root)
		{
   
			_root = subR;
			_root->_parent = nullptr;
		}
		else
		{
   
			if (parent == ppNode->_left)
			{
   
				ppNode->_left = subR;
			}
			else
			{
   
				ppNode->_right = subR;
			}

			subR->_parent = ppNode;
		}
	}

	void RotateR(Node* parent)
	{
   
		Node* subL = parent->_left;
		Node* subLR = subL->_right;

		parent->_left = subLR;
		if (subLR)
			subLR->_parent = parent;

		Node* ppNode = parent->_parent;

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