数据结构 --- 红黑树基础

什么是红黑树？

红黑树(Red-Black Tree，简称R-B Tree)，它是一种特殊的二叉搜索树。意味着它满足二叉搜索树的特征：
任意一个节点所包含的键值，大于等于左孩子的键值，小于等于右孩子的键值。
除了具备该特性之外，红黑树还包括许多额外的信息。

红黑树的每个节点上都有存储位表示节点的颜色，颜色是红(Red)或黑(Black)。
红黑树的特性:
(1) 每个节点或者是黑色，或者是红色。
(2) 根节点是黑色。  如果插入的时候发现根节点变成红色需要进行树的调整
(3) 每个叶子节点是黑色。 [注意：这里叶子节点，是指 为空的 叶子节点！即最后一个节点的孩子节点]
(4) 如果一个节点是红色的，则它的子节点必须是黑色的。
(5) 从一个节点到该节点的子孙节点的所有路径上包含相同数目的黑节点。

关于它的特性，需要注意的是：
第一，特性(3)中的叶子节点，是只为空(NIL或null)的节点。
第二，特性(5)，确保没有一条路径会比其他路径长出俩倍。
因而，红黑树是相对是接近平衡的二叉树。

红黑树示意图如下：

定理：一棵含有n个节点的红黑树的高度至多为2lg(n+1)

红黑树的基本操作是添加、删除和旋转。
在对红黑树进行添加或删除后，会用到旋转方法。为什么呢？道理很简单，添加或删除红黑树中的节点之后，红黑树就发生了变化，可能不满足红黑树的5条性质，也就不再是一颗红黑树了，而是一颗普通的树。而通过旋转，可以使这颗树重新成为红黑树。
简单点说，旋转的目的是让树保持红黑树的特性。
旋转包括两种：左旋 和 右旋。

红黑树左旋示意图如下：

对x进行左旋，意味着"将x变成一个左节点"。

红黑树右旋示意图如下：

对y进行左旋，意味着"将y变成一个右节点"。

红黑树的插入

根据被插入节点的父节点的情况，可以将"当节点z被着色为红色节点，并插入二叉树"划分为三种情况来处理。
① 情况说明：被插入的节点是根节点。
     处理方法：直接把此节点涂为黑色。（红黑树的根节点必须是黑色）
② 情况说明：被插入的节点的父节点是黑色。
     处理方法：什么也不需要做。节点被插入后，仍然是红黑树。
③ 情况说明：被插入的节点的父节点是红色。
     处理方法：那么，该情况与红黑树的“特性(5)”相冲突。
这种情况下，被插入节点是一定存在非空祖父节点的；
进一步的讲，被插入节点也一定存在叔叔节点
(即使叔叔节点为空，我们也视之为存在，空节点本身就是黑色节点)。
理解这点，我们依据"叔叔节点的情况"，将这种情况进一步划分为3种情况

被插入的节点的父节点是红色的处理策略

No.1 叔叔是红色
当前节点(即，被插入节点)的父节点是红色，
且当前节点的祖父节点的另一个子节点（叔叔节点）也是红色

处理策略
1.将“父节点”设为黑色。
2.将“叔叔节点”设为黑色。
3.将“祖父节点”设为“红色”。
4.将“祖父节点”设为“当前节点”(红色节点)；继续对“当前节点”进行操作。

将“祖父节点”设为“当前节点”(红色节点)；继续对“当前节点”进行操作

No.2 叔叔是黑色，且当前节点是右孩子
当前节点(即，被插入节点)的父节点是红色，叔叔节点是黑色，
且当前节点是其父节点的右孩子

处理策略
1.将“父节点”作为“新的当前节点”。
2.以“新的当前节点”为支点进行左旋

No.3 叔叔是黑色，且当前节点是左孩子
当前节点(即，被插入节点)的父节点是红色，叔叔节点是黑色，
且当前节点是其父节点的左孩子

处理策略
1.将“父节点”设为“黑色”。
2.将“祖父节点”设为“红色”。
3.以“祖父节点”为支点进行右旋。

红黑树的删除

红黑树的几个特性：
(1) 每个节点或者是黑色，或者是红色。
(2) 根节点是黑色。
(3) 每个叶子节点是黑色。 [注意：这里叶子节点，是指为空的叶子节点！]
(4) 如果一个节点是红色的，则它的子节点必须是黑色的。
(5) 从一个节点到该节点的子孙节点的所有路径上包含相同数目的黑节点。

先做删除，再做调整，可能会违背：

特性(2)删除根节点可能会把一个红节点挪上去作为根节点

特性(4)删除后把一个节点挪上去会存在红红现象

特性(5)如果把黑节点删掉会导致这一条线上的黑节点数减少，需要去做调整

将红黑树内的某一个节点删除。需要执行的操作依次是：
第一步:将红黑树当作一颗二叉查找树，将该节点从二叉查找树中删除；
第二步:通过"旋转和重新着色"等一系列来修正该树，使之重新成为一棵红黑树。
详细描述如下：
第一步：将红黑树当作一颗二叉查找树，将节点删除。
这和"删除常规二叉查找树中删除节点的方法是一样的"。分3种情况：
① 被删除节点没有儿子，即为叶节点。那么，直接将该节点删除就OK了。
② 被删除节点只有一个儿子。直接删除，用该节点的唯一子节点顶替它的位置。
③ 被删除节点有两个儿子。那么，先找出它的后继节点；
    然后把“它的后继节点的内容”复制给“该节点的内容”；
    之后，删除“它的后继节点”。再将后继节点删除。
   "被删除节点"有两个非空子节点的情况下，它的后继节点不可能是双子非空。
     意味着"该节点的后继节点"要么没有儿子，要么只有一个儿子。
    若没有儿子，则按"情况① "进行处理；
    若只有一个儿子，则按"情况② "进行处理。

第二步：通过"旋转和重新着色"等一系列来修正该树，使之重新成为一棵红黑树。
因为"第一步"中删除节点之后，可能会违背红黑树的特性。所以需要通过"旋转和重新着色"来修正该树，使之重新成为一棵红黑树。

将"删除红黑树中的节点"大致分为两步，
NO.1:"当作一颗二叉查找树，将节点删除"后，可能违反"特性(2)、(4)、(5)"三个特性。
NO.2:需要解决上面的三个问题，进而保持红黑树的全部特性。
为了便于分析，我们假设"x包含一个额外的黑色"(x原本的颜色还存在)，这样就不会违反"特性(5)"。
为什么呢？通过RB-DELETE算法，我们知道：删除节点y之后，x占据了原来节点y的位置。
既然删除y(y是黑色)，意味着减少一个黑色节点；那么，再在该位置上增加一个黑色即可。
这样，当我们假设"x包含一个额外的黑色"，就正好弥补了"删除y所丢失的黑色节点"，也就
不会违反"特性(5)"。 因此，假设"x包含一个额外的黑色"(x原本的颜色还存在)，这样就不会
违反"特性(5)"。现在，x不仅包含它原本的颜色属性，x还包含一个额外的黑色。即x的颜色属性
是"红+黑"或"黑+黑"，它违反了"特性(1)"。现在，由解决"违反了特性(2)、(4)、(5)三个特性
"转换成了"解决违反特性(1)、(2)、(4)三个特性"。删除函数需要做的就是通过算法恢复红黑树
的特性(1)、(2)、(4)。
删除调整的思想是：将x所包含的额外的黑色不断沿树上移(向根方向移动)，直到出现下面的姿态：
1.x指向一个"红+黑"节点。此时，将x设为一个"黑"节点即可。
2.x指向根。此时，将x设为一个"黑"节点即可。
3.非前面两种姿态。

将上面的姿态，可以概括为3种情况。
① 情况说明：x是“红+黑”节点。
    处理方法：直接把x设为黑色，结束。此时红黑树性质全部恢复。
② 情况说明：x是“黑+黑”节点，且x是根。
    处理方法：什么都不做，结束。此时红黑树性质全部恢复。
③ 情况说明：x是“黑+黑”节点，且x不是根。
    处理方法：这种情况又可以划分为4种子情况。这4种子情况如下表所示：

No.1 x是"黑+黑"节点，x的兄弟节点是红色

处理策略
1.将x的兄弟节点设为“黑色”。
2.将x的父节点设为“红色”。
3.对x的父节点进行左旋。
4.左旋后，重新设置x的兄弟节点。

这样做的目的是将“No.1”转换为“No.2”、“No.3”或“No.4”，
从而进行进一步的处理。对x的父节点进行左旋；左旋后，为了保持红黑树特性，
就需要在左旋前“将x的兄弟节点设为黑色”，同时“将x的父节点设为红色”；
左旋后，由于x的兄弟节点发生了变化，需要更新x的兄弟节点，从而进行后续处理。

No.2 x是"黑+黑"节点，x的兄弟节点是黑色，x的兄弟节点的两个孩子都是黑色

处理策略
1.将x的兄弟节点设为“红色”。
2.设置“x的父节点”为“新的x节点”

将“x中多余的一个黑色属性上移(往根方向移动)”。 x是“黑+黑”节点，我们将x由“黑+黑”节点 变成 “黑”节点，多余的一个“黑”属性移到x的父节点中，即x的父节点多出了一个黑属性(若x的父节点原先是“黑”，则此时变成了“黑+黑”；若x的父节点原先时“红”，则此时变成了“红+黑”)。 此时，需要注意的是：所有经过x的分支中黑节点个数没变化；但是，所有经过x的兄弟节点的分支中黑色节点的个数增加了1(因为x的父节点多了一个黑色属性)！我们需要将“所有经过x的兄弟节点的分支中黑色节点的个数减1”即可，那么就可以通过“将x的兄弟节点由黑色变成红色”来实现。
经过上面的步骤(将x的兄弟节点设为红色)，多余的一个颜色属性(黑色)已经跑到x的父节点中。我们需要将x的父节点设为“新的x节点”进行处理。若“新的x节点”是“黑+红”，直接将“新的x节点”设为黑色，即可完全解决该问题；若“新的x节点”是“黑+黑”，则需要对“新的x节点”进行进一步处理。

 No.3 x是"黑+黑"节点，x的兄弟节点是红色

 处理策略
1.将x兄弟节点的左孩子设为“黑色”。
2.将x兄弟节点设为“红色”。
3.对x的兄弟节点进行右旋。
4. 右旋后，重新设置x的兄弟节点。

我们处理“No.1”的目的是为了将“No3”进行转换，转换成“No.4”,
从而进行进一步的处理。转换的方式是对x的兄弟节点进行右旋；
为了保证右旋后，它仍然是红黑树，就需要在右旋前
“将x的兄弟节点的左孩子设为黑色”，
“将x的兄弟节点设为红色”；
右旋后，由于x的兄弟节点发生了变化，需要更新x的兄弟节点，从而进行后续处理。

No.4x是“黑+黑”节点，x的兄弟节点是黑色；
x的兄弟节点的右孩子是红色的，x的兄弟节点的左孩子任意颜色

处理策略
1.将x父节点颜色 赋值给 x的兄弟节点。
2.将x父节点设为“黑色”。
3.将x兄弟节点的右子节设为“黑色”。
4.对x的父节点进行左旋。
5.设置“x”为“根节点”。

 着色用的

#include <stdio.h>
#include <stdlib.h>
#define RED 0
#define BLACK 1

 红黑树节点结构体

typedef struct RBTreeNode
{
	//size_t color;
	unsigned int color;                //颜色
	int key;                           //键
	struct RBTreeNode* LChild;         //左子树指针
	struct RBTreeNode* RChild;         //右子树指针
	struct RBTreeNode* parentNode;     //父节点
}NODE,*LPNODE;

 再封装红黑树结构体 

typedef struct RBTree 
{
	LPNODE root;                        //用一个节点表示整棵树
	int treeSize;                       //树的大小 | 当前元素个数
}RBTREE,*LPRBTREE;

 左旋 

//针对哪棵树做的旋转 针对哪个节点
void L_Rotation(LPRBTREE tree, LPNODE x) 
{
	//1.拿x节点的右边y
	LPNODE y = x->RChild;
	//2.把y原来的左边β成为x的右边->先把y的左边β拿出来才能修改y的左边 如果直接连接xy 会覆盖β
	x->RChild = y->LChild;
    //判断x有没有父节点,如果x没有父节点,y就需要成为根节点
    //注意:只是处理了x指向β,但是没有处理β的parentNode属性,引入了parentNode需要对它做处理
	//3.处理连接后的节点中parentNode
	if (y->LChild != NULL) 
	{
		y->LChild->parentNode = x;      //如果原来y节点的左子树β!=NULL,需要改变β的父节点
	}
	//y既然替换了x,y的父节点是x的父节点
	y->parentNode = x->parentNode;
	//x的父节点也要改
	if (x->parentNode == NULL)          //原来x不存在父节点，x原来是根节点
	{
		tree->root = y;                 //修改根节点为y
	}
	else 
	{
		//判断: 是把y放在原来x的父节点的左边还是右边->让y替换x的作用
		if (x->parentNode->LChild == x) 
		{
			x->parentNode->LChild = y;
		}
		else 
		{
			x->parentNode->RChild = y;
		}
	}
	//最后处理x的父节点
	y->LChild=x;                         //y的左子树指向父节点
	x->parentNode = y;	                 //x的父节点改成y
}

 右旋 

void R_Rotation(LPRBTREE tree,LPNODE y)
{
	//拿到y节点的左边x
	LPNODE x = y->LChild;
	y->LChild = x->RChild;
	if (x->RChild != NULL) 
	{
		x->RChild->parentNode = y;
	}
	//处理x和y的parentNode
	x->parentNode = y->parentNode;
	if (y->parentNode == NULL) 
	{
		tree->root = x;
	}
	else 
	{
		if (y == y->parentNode->LChild) 
		{
			y->parentNode->LChild = x;
		}
		else 
		{
			y->parentNode->RChild = x;
		}
	}
	//处理x的右边和y的父节点
	x->RChild = y;
	y->parentNode = x;
}