红黑树详解

1 红黑树的应用

1.Linux进程调度CFS
2.Nginx Timer事件管理
3.Epoll事件块的管理

2 红黑树概念

• 每个节点都是红的或者是黑色的
• 根节点是黑色的
• 每个叶子节点是黑色的
• 如果一个节点是红色的，则它的两个子节点是黑色的
• 从任意节点到其每个叶子的所有路径都包含相同数量的黑色节点

3 红黑树实现代码(C++)

3.1 红黑树节点实现

typedef int KEY_TYPE;

typedef struct _rbtree_node {
	
	KEY_TYPE key;
	void *value;

	struct _rbtree_node *right;
	struct _rbtree_node *left;
	struct _rbtree_node *parent;
	unsigned char color;

} rbtree_node;

3.2 红黑树实现

typedef struct _rbtree {
	
	struct _rbtree_node *root;
	struct _rbtree_node *nil;

} rbtree;

3.3 左旋和右旋

3.3.1 左旋代码实现

void rbtree_left_roate(rbtree *T, rbtree_node *x) {	// 左旋

	rbtree_node *y = x->right;

	// 1.
	x->right = y->left;
	if (y->left != T->nil) {	// 由于叶子节点的parent,left,right指针都指向NIL节点,所以是叶子节点的话不执行操作
		y->left->parent = x;
	}

	// 2.
	y->parent = x->parent;
	if(x->parent == T->nil) {			// 因为根节点的parent指向NIL节点,所以说明x是根节点,根节点变为y
		T->root = y;
	} else if (x == x->parent->left) {	// 不是根节点 则判断x为x->parent的左节点还是右节点
		x->parent->left = y;
	} else {
		x->parent->right = y;
	}

	// 3.
	y->left = x;
	x->parent = y;
}

3.3.2右旋代码实现

右旋和左旋类似，具有对称性和可逆性，所以代码可以将x与y互换，left与right互换

void rbtree_right_roate(rbtree *T, rbtree_node *y) {	// 右旋

	rbtree_node *x = y->left;

	// 1.
	y->left = x->right;
	if (x->right != T->nil) {	// 叶子节点的parent,left,right指针都指向NIL节点,所有叶子是节点的话不执行操作
		x->right->parent = y;
	}

	// 2.
	x->parent = y->parent;
	if(y->parent == T->nil) {			// y是根节点 根节点变为x
		T->root = x;
	} else if (y == y->parent->right) {	// 不是根节点 则判断y是y->parent的左节点还是右节点
		y->parent->right = x;
	} else {
		y->parent->left = x;
	}

	// 3.
	x->right = y;
	y->parent = x;
}

3.4 插入操作

void rbtree_insert(rbtree *T, rbtree_node *z) { // z: 插入的节点

	rbtree_node *y = T->nil; // 用来存储nil节点的上一个节点，因为下面while遍历得到的是nil节点没用
	rbtree_node *x = T->root;

	while(x != T->nil) {
		y = x;
		if(z->key < x->key) {	
			x = x->left;
		} else if (z->key > x->key) {
			x = x->right;
		} else { // Exist 看具体业务
			return ;
		}

	}

	if(y == T->nil) {		// 该树没有节点
		T->root = z;
	} else {
		if(z->key < y->key) {
			y->left = z;
		} else {
			y->right = z;
		}
	}

	z->parent = y;

	z->left = T->nil;
	z->right = T->nil;
	z->color = RED;				// 新插入的节点定为红色的好处是不影响原来红黑色的黑高(各路径黑高相同)

	rbtree_insert_fixup(T, z);	// 再进行调整
}

3.5 调整函数

• 情况1
  
• 情况2
  
• 情况3
  

下面的情况4、5、6与1、2、3是对称关系，只需把左旋操作与右旋操作互换即可，具体可以自己将情况1、2、3的图示对称就能形成4、5、6的情况

void rbtree_insert_fixup(rbtree *T, rbtree_node *z) {

	// z == RED 因为新插入节点z默认红色(不改变红黑树黑高)
	while(z->parent->color == RED) {

		if(z->parent == z->parent->parent->left) { 		// z的父节点是z的祖父节点的左子树
			rbtree_node *y = z->parent->parent->right; 	// 得到z的叔父节点

			// 情况1
			if(y->color == RED) {

				z->parent->color = BLACK;
				y->color = BLACK;
				z->parent->parent->color = RED;
				
				z = z->parent->parent;	// z的祖父节点变成红色，所以继续往上调整，z回溯(每次回溯z都是红色)

			} 
			// 情况1
			else 
			{	// y == BLACK 即叔父节点为黑色
				
				// 情况3
				if(z == z->parent->right) { // ppt16页 

				z = z->parent;
				rbtree_left_rotate(T, z);

				}

				// 情况2
				z->parent->color = BLACK;
				z->parent->parent->color = RED;

				rbtree_right_rotate(T, z->parent->parent);
				// 情况2

				// 情况3
			}

		// 下面的三种情况与上面三种是对称的操作，把左旋换成右旋即可(可以把上面三种情况图对称画出来就是下面的三种情况)
		} else {	// z的父节点是z的祖父节点的右子树

			rbtree_node *y = z->parent->parent->left;	// 得到z的叔父节点
			
			// 情况4 	和情况1一样
			if(y->color == RED) {

				z->parent->color = BLACK;
				y->color = BLACK;
				z->parent->parent->color = RED;

				z = z->parent->parent;	// z的祖父节点变成红色，所以继续往上调整，z回溯(每次回溯z都是红色)

			} 
			// 情况4	
			else {	// y == BLACK 即叔父节点为黑色 

				// 情况6	情况6的对称
				// 情况5	情况5的对称
				if(z == z->parent-left) {
					z = z->parent;
					rbtree_right_rotate(T, z);
				}
				// 情况5

				z->parent->color = BLACK;
				z->parent->parent->color = RED;
				rbtree_left_rotate(T, z->parent->parent);
				// 情况6
			}

		}

	}
}