理论知识参见 《数据结构基础 》 张力译版以下代码与大家分享,如有错误之处希望批评指正。(我已测试通过)
//RedBlackTree.h
#ifndef _REDBLACKTREE_
#define _REDBLACKTREE_
class TreeNode
{
friend class RedBlackTree;
public:
TreeNode(int e)
{
key=e;
rb=0;
leftchild=rightchild=0;
}
private:
int key;
bool rb;//颜色标记0 红 1 black
TreeNode *leftchild, *rightchild;
};
class RedBlackTree
{
public:
RedBlackTree(TreeNode* init=0)
{
root=0;
}
void Insert(const int e);
private:
TreeNode * root;
};
#endif
//RedBlackTree.cpp
#include "RedBlackTree.h"
#include <iostream>
#include <stack>
using namespace std;
void RedBlackTree::Insert(const int e)
{
if (!root)
{
root=new TreeNode(e);
root->rb=1;//根节点始终是黑色
return ;
}
stack<TreeNode*>s;//记录遍历路径
TreeNode *u=root;
TreeNode *pu=0;//u的父节点
TreeNode *gu=0;//u的祖父节点
TreeNode *uu=0;//叔父节点
while(u)//查找插入位置
{
s.push(u);
/*关键字查找*/
if (e==u->key)//已有该关键字
{
return;
}
else if (e<u->key)
{
u=u->leftchild;
}
else
{
u=u->rightchild;
}
}
bool inflag=0; //新节点尚未插入
u=new TreeNode(e);
pu=s.top();//得到父节点
s.pop();
while (1)//一直进行
{
if (s.empty())//s空 仅有父节点
{
//无祖父节点
if (!inflag)// 尚未插入
{
if (e<pu->key)
{
pu->leftchild=u;
}
else if (e>pu->key)
{
pu->rightchild=u;
}
inflag=1;
}
return ;
}
else
{
gu=s.top();
s.pop();
//尚未插入则插入
if (!inflag)// 尚未插入
{
if (e<pu->key)
{
pu->leftchild=u;
}
else if (e>pu->key)
{
pu->rightchild=u;
}
inflag=1;
}
//判断当前的u是否违反性质2 不违反则返回,违反则分类操作
if (pu->rb!=u->rb)//不违反性质2
{
return;
}
else
{
//分类 r b
if (pu==gu->leftchild)
{
uu=gu->rightchild;
}
else
{
uu=gu->leftchild;
}
if (uu&&uu->rb==0)//red 颜色变换 LL 与RL 变换相同
{
if (gu->leftchild==pu)//1L
{
// if (pu->leftchild==u)//2L (LL)
// {
//LLr 颜色变换
gu->rb=0;
uu->rb=1;
pu->rb=1;
// }
// else//2R (LR)
// {
// gu->rb=0;
// uu->rb=1;
// pu->rb=1;
// }
}
else//1R
{
// if (pu->leftchild==u)//2L (RL)
// {
//RLr 颜色变换
gu->rb=0;
uu->rb=1;
pu->rb=1;
// }
// else//2R (RR)
// {
// gu->rb=0;
// uu->rb=1;
// pu->rb=1;
// }
}
u=gu;//准备下次判断、操作
if (!s.empty())
{
pu=s.top();
s.pop();
}
else
{
u->rb=1;//root
}
}
else// b 旋转
{
if (gu->leftchild==pu)//1L
{
if (pu->leftchild==u)//2L (LL)
{
//LLb 旋转
gu->leftchild=pu->rightchild;
pu->rightchild=gu;
pu->rb=1;
gu->rb=0;
}
else//2R (LRb)
{
gu->leftchild=u->rightchild;
pu->rightchild=u->leftchild;
u->leftchild=pu;
u->rightchild=gu;
u->rb=1;
gu->rb=0;
pu->rb=0;
}
}
else//1R
{
if (pu->leftchild==u)//2L (RL)
{
//RLb 旋转
pu->leftchild=u->rightchild;
gu->rightchild=u->leftchild;
u->rightchild=pu;
u->leftchild=gu;
u->rb=1;
gu->rb=0;
pu->rb=0;
}
else//2R (RR)
{
gu->rightchild=pu->leftchild;
pu->leftchild=gu;
pu->rb=1;
gu->rb=0;
}
}
if (!s.empty())
{
pu=s.top();
s.pop();
}
else
{
root=u;
u->rb=1;//root
return;
}
if (e<pu->key)
{
pu->leftchild=u;
}
else
{
pu->rightchild=u;
}
return;
}
}
}
}
}
//main.cpp
#include <iostream>
#include "RedBlackTree.h"
using namespace std;
void main()
{
RedBlackTree rbt;
rbt.Insert(50);
rbt.Insert(10);
rbt.Insert(80);
rbt.Insert(90);
rbt.Insert(70);
rbt.Insert(60);
rbt.Insert(65);
rbt.Insert(62);
}