typedef struct node{
int key,color;
struct node *lchild;
struct node *rchild;
struct node *parent;
}Pnode;
结构体类型就是在二叉搜索树的基础上增加了color这个变量
插入操作:
插入的元素默认为红色
除了基本的二叉树元素插入之外还增加了红黑调整满足红黑树的五大性质
假设插入节点为X,无论哪种插入情况只会影响性质2和性质4其中的一个.
1.当红黑树为空树时,此时只需将插入的元素作为根节点并把颜色改成黑色即可
2.当红黑树不为空时,当且仅当插入后,X的父节点与X一样为红色时,违反性质4。
一.当X的叔叔节点(即X的父亲的兄弟节点)为红色时, 将X的爷爷节点改为红色,将X的父节点和叔叔节点改为黑色,并将X的爷爷节点进 入下一次调整。
二.当X的叔叔节点为黑色时,如果X为其父节点的左孩子,将其父节点改为黑色,将爷爷节点改成红色并对爷爷节点进行右旋。
三.当X的叔叔节点为黑色时,如果X为其父节点的右孩子,对其父节点进行左旋,再进行第二种情况操作。
代码:
//插入
void insert(Pnode **root,int x)
{
Pnode *p=(Pnode *)malloc(sizeof(Pnode));
p->lchild=p->rchild=p->parent=NULL;
p->color=red;p->key=x;
//如果为空树 违反性质2 因此直接将红色改成黑色即可
if(*root==NULL) {
p->color=black;
*root=p;
return;
}
if((*root)->rchild==NULL&&(*root)->key<x){
(*root)->rchild=p;
p->parent=*root;
insertfix(&p);
return;
}
if((*root)->lchild==NULL&&(*root)->key>x){
(*root)->lchild=p;
p->parent=*root;
insertfix(&p);
return;
}
if((*root)->key<x){
insert(&(*root)->rchild,x);
}
else {
insert(&(*root)->lchild,x);
}
}
//插入调整
void insert(Pnode **root,int x)
{
Pnode *p=(Pnode *)malloc(sizeof(Pnode));
p->lchild=p->rchild=p->parent=NULL;
p->color=red;p->key=x;
//如果为空树 违反性质2 因此直接将红色改成黑色即可
if(*root==NULL) {
p->color=black;
*root=p;
return;
}
if((*root)->rchild==NULL&&(*root)->key<x){
(*root)->rchild=p;
p->parent=*root;
insertfix(&p);
return;
}
if((*root)->lchild==NULL&&(*root)->key>x){
(*root)->lchild=p;
p->parent=*root;
insertfix(&p);
return;
}
if((*root)->key<x){
insert(&(*root)->rchild,x);
}
else {
insert(&(*root)->lchild,x);
}
}