算法 红黑树_红黑树的具体算法

红黑树概述

红黑树（Red Black Tree）是一种自平衡二叉查找树，是在计算机科学的中用到的一种数据结构，典型的用途是实现关联数组，红黑树和AVL树类似，都是在进行插入和删除操作时通过特定操作保持二叉查找树的平衡，从而获得较高的查找性能

它虽然是复杂到，但它的最坏情况运行时间也是非常良好的，并且在实践中是高效的：它可以在O(log n)时间内做查找，插入和删除，这里的n是树中元素的数目

红黑树性质

红黑树是每个结点都带有颜色属性的二叉查找树，颜色是红色或黑色，在二叉查找树强制一般要求以外，对于任何有效的红黑树我们增加了如下的额外要求：

• 性质1 每个结点是红色或黑色
• 性质2 根节点是黑色
• 性质3 每个叶结点（NIL）是黑色
• 性质4 每个红色结点的两个子结点都是黑色（从每个叶子到根的所有路径上不能有两个连续的红色结点）
• 性质 5 从任一结点到其每个叶子的所有路径都包含相同数目的黑色节点

这些约束强制了红黑树的关键性质：从根到叶子的最长的可能路径不多与最短的可能路径的两倍长，结果是这个树大致是平衡的，因为操作比如插入、删除和查找某个值的最坏情况时间都要求与树的高度成正比例，这个在高度上的理论上限允许红黑树在最坏情况下都是高效的，而不同于普通的二叉查找树

是性质 3 导致路径上不能有两个连续的红色节点确保了这个结果，最短的可能路径都是黑色节点，最长的可能路径有交替的红色和黑色节点，因为根据性质 4 所有最长的路径都有相同数目的黑色节点，这就表明了没有路径能多于任何其他路径的两倍长

红黑树的插入

代码实现

红黑树的左右单旋的代码与AVL树是一致的，根据上面的规则来实现红黑树的插入

typedef enum { RED = 0, BLACK = 1 }ColorType;

typedef int KeyType;
typedef struct rb_node
{
	rb_node* leftchild;
	rb_node* parent;
	rb_node* rightchild;
	ColorType color; //AVL int balance
	KeyType key;
}rb_node,*RBTree;

rb_node* Buynode();
static rb_node* Nil = Buynode(); //哨兵节点

rb_node* Buynode()
{
	rb_node* s = (rb_node*)malloc(sizeof(rb_node));
	if (s == nullptr) exit(1);
	memset(s, 0, sizeof(rb_node));
	s->rightchild = Nil;
	s->leftchild = Nil; //新节点左右孩子都指向哨兵
	s->color = RED;
	return s;
}
rb_node* MakeRoot(KeyType kx)    //建根
{
	rb_node* root = Buynode();
	root->color = BLACK;
	root->key = kx;
	return root;
}

void RotateLeft(rb_node*& tree, rb_node* ptr)
{
	rb_node* newroot = ptr->rightchild;
	newroot->parent = ptr->parent;

	if (newroot->leftchild != nullptr)
	{
		newroot->leftchild->parent = ptr;
	}
	ptr->rightchild = newroot->leftchild;

	newroot->leftchild = ptr;
	if (ptr == tree)
	{
		tree = newroot;
	}
	else
	{
		if (ptr->parent->leftchild == ptr)
		{
			ptr->parent->leftchild = newroot;
		}
		else
		{
			ptr->parent->rightchild = newroot;
		}
	}

	ptr->parent = newroot;
}
void RotateRight(rb_node*& tree, rb_node* ptr)
{
	rb_node* newroot = ptr->leftchild;
	newroot->parent = ptr->parent;
	ptr->leftchild = newroot->rightchild;
	if (newroot->rightchild != nullptr)
	{
		newroot->rightchild->parent = ptr;
	}

	newroot->rightchild = ptr;
	if (ptr == tree)
	{
		tree = newroot;
	}
	else
	{
		if (ptr->parent->leftchild == ptr)
		{
			ptr->parent->leftchild = newroot;
		}
		else
		{
			ptr->parent->rightchild = newroot;
		}
	}

	ptr->parent = newroot;
}

void PassRBTree(rb_node*& tree, rb_node* p)
{
	rb_node* _X = nullptr;
	for (; p != tree && p->parent->color == RED;)
	{
		if (p->parent->parent->rightchild == p->parent) //is right
		{
			_X = p->parent->parent->leftchild;
			if (_X->color == RED)
			{
				_X->color = BLACK;
				p->parent->color = BLACK;
				p->parent->parent->color = RED;
				p = p->parent->parent;
			}
			else
			{
				if (p->parent->leftchild == p) //我爹在双亲右边，我在双亲左边，是个折线
				{
					p = p->parent;
					RotateRight(tree, p); //先右单旋
				}
				p->parent->color = BLACK;
				p->parent->parent->color = RED;
				RotateLeft(tree, p->parent->parent); //左单旋转

			}
		}
		else
		{
			_X = p->parent->parent->rightchild;
			if (_X->color == RED)
			{
				_X->color = BLACK;
				p->parent->color = BLACK;
				p->parent->parent->color = RED;
				p = p->parent->parent;
			}
			else
			{
				if (p->parent->rightchild == p) 
				{
					p = p->parent;
					RotateLeft(tree, p);
				}
				p->parent->color = BLACK;
				p->parent->parent->color = RED;
				RotateRight(tree, p->parent->parent); 

			}
		}
	}
	tree->color = BLACK;
}

bool Insert(rb_node*& tree, KeyType kx)
{
	if (tree == nullptr)
	{
		tree = MakeRoot(kx);
		return true;
	}
	rb_node* pa = nullptr;
	rb_node* p = tree;
	while (p != nullptr && p->key != kx)
	{
		pa = p;
		p = kx < p->key ? p->leftchild : p->rightchild;
	}
	if (p != nullptr && p->key == kx) return false;
	p = Buynode();
	p->key = kx;
	p->parent = pa;
	if (kx < pa->key)
	{
		pa->leftchild = p;
	}
	else
	{
		pa->rightchild = p;
	}
	PassRBTree(tree, p); //回溯
	return true;
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177