标签:
接下来介绍红黑树的插入操作,介绍插入之前,我们先来了解一下红黑树的性质。
1、每个节点不是红色就是黑色
2、跟节点为黑色。
3、如果节点为红,子节点必须为黑。
4、任意节点至树尾端的任何路径,黑节点必须相同。
规则4主要是保证树的平衡性,不过它的要求不是很严。主要是为了减少调整操作。根据规则4,我们可以判断出新节点都是红节点。(如果新节点是黑节点,那么每次插入都要进行调整)
由于要经常使用某个节点的父亲。所以这里添加了一个指向父亲的指针。所以我们先看一下红黑树的结构组成。
typedef int value_type; typedef bool color_type; const color_type red = true; const color_type black = false; typedef struct node { value_type value; color_type color; struct node * parent; struct node * left; struct node * right; } Node; typedef struct Tree { Node * root; } Tree;
Node节点,value,关键字信息。color,颜色。parent,left,right,父亲,左孩子,右孩子。
Tree只有一个根节点。
插入操作,与二叉树的插入是一样的。
首先是找到插入位置。
//根节点返回NULL,找到返回当前节点,否则返回x的父亲(x是节点所在的位置) static Node * search_node(const int key, Tree * tree) { Node * x, * y; x = tree->root; y = nil; while ( x != nil ) { y = x; if ( key < x->value ) x = x->left; else if ( key > x->value ) x = x->right; else break; } return y; }
nil是一个空节点,所有应该为空的指针都指向它。这个节点是黑色的,parent,left,right都为NULL,没有初始化value。(不是一定要有nil节点)
其次将元素插入
void rb_insert(const value_type value, Tree * tree) { Node * new_node;//新节点 Node * x = search_node(value, tree);//插入点 if ( x == nil ) { tree->root = make_new(value); tree->root->color = black;//根节点为黑色 } else { if ( x->value == value )//关键字不可重复 return; new_node = make_new(value); if ( value < x->value )//将新节点插入二叉树中 x->left = new_node; else x->right = new_node; new_node->parent = x;//设置新节点的父亲 if ( x->color == red )//如果新节点的父亲为红色,调整 insert_fixup(new_node, tree); } }
新节点一定是红色的,所以不会违反规则4,但有可能违反规则3。也就是说如果插入节点的父亲是红色的,那么违反了规则3。
我们需要进行调整。
调整时会遇到三种情况,根据不同的情况进行不同的调整。
调整是一个循环过程,当x为根节点或者x是黑色时结束。(这个操作次数并不会很多)
三种情况在代码中看注释。
static void insert_fixup(Node * x, Tree * tree) { Node * y;//y是x的伯父节点 while ( x != tree->root && x->parent->color == red ) { if ( x->parent == x->parent->parent->left )//父亲是祖父左孩子 { y = x->parent->parent->right; //case 1: 伯父是红色 //处理办法 // 1.将父亲,伯父变成黑色 // 2.将祖父变成红色 // 3.将祖父设为x,向上检查 if ( y->color == red ) { x->parent->color = black; y->color = black; x->parent->parent->color = red; x = x->parent->parent; } else { //case 2: x是父亲的右孩子 //处理办法: // 1.将x设为x的父亲 // 2.以x为旋转点,左旋 //经过上述操作,转换成case 3 if ( x == x->parent->right ) { x = x->parent; left_rotate(x, tree); } //case 3: x是父亲的左孩子 //处理办法: // 1.将x的父亲设为黑色 // 2.将x的祖父设为红色 // 3.以x的祖父为旋转点,右旋 x->parent->color = black; x->parent->parent->color = red; right_rotate(x->parent->parent, tree); } } else//与上述处理相同,只是将left与right互换 { y = x->parent->parent->left; if ( y->color == red ) { x->parent->color = black; y->color = black; x->parent->parent->color = red; x = x->parent->parent; } else { if ( x == x->parent->left ) { x = x->parent; right_rotate(x, tree); } x->parent->color = black; x->parent->parent->color = red; left_rotate(x->parent->parent, tree); } } }//while end tree->root->color = black; }
这样插入操作就完成了。
这是一棵红黑树,圆圈里的代表黑色节点,没有圆圈的代表红色。大家可以利用这棵树来调代码。建议在调整时,用printf把case 1到3标记一下,挨个试验。
这颗树的插入顺序是10,7,8,15,5,6,11,13,12。
标签:
原文地址:http://www.cnblogs.com/ITgaozy/p/5180444.html