RB Tree的部分实现

最新推荐文章于 2024-02-25 22:14:15 发布

zk_sima

最新推荐文章于 2024-02-25 22:14:15 发布

阅读量595

点赞数

分类专栏： STL学习文章标签： tree insert c

本文链接：https://blog.csdn.net/zk_sima/article/details/6277692

版权

STL学习专栏收录该内容

13 篇文章 0 订阅

订阅专栏

跟前面一样，主要是平衡度的问题，RB树是通过节点上的颜色来识别是否平衡的。那么关键是怎么知道是否平衡，怎么恢复平衡。

首先每个节点有一个成员用于记录节点的颜色：

默认节点的颜色为红色，因为插入的叶子节点都是红色的。

是否平衡通过判断。插入的节点和其父节点的颜色来比较。

恢复平衡跟前面一样。通过旋转得到，单旋（左旋，右旋），双选（左旋+右旋，右旋+左旋）：

private://负责执行插入，删除，维持平衡的实体 void l_rotaion(PNODE p_child_node,PNODE p_parent_node) { p_parent_node->r_child=p_child_node->l_child; if(p_child_node->l_child!=0) p_child_node->l_child->parent=p_parent_node; if(p_parent_node==head->parent) { head->parent=p_child_node; } else { if(p_parent_node==p_parent_node->parent->l_child) { p_parent_node->parent->l_child=p_child_node; } else { p_parent_node->parent->r_child=p_child_node; } } p_child_node->parent=p_parent_node->parent; p_child_node->l_child=p_parent_node; p_parent_node->parent=p_child_node; } void r_roation(PNODE p_child_node,PNODE p_parent_node) { p_parent_node->l_child=p_child_node->r_child; if(p_child_node->r_child!=0) p_child_node->r_child->parent=p_parent_node; if(p_parent_node==head->parent) { head->parent=p_child_node; } else { if(p_parent_node==p_parent_node->l_child) { p_parent_node->parent->l_child=p_child_node; } else { p_parent_node->parent->r_child=p_child_node; } } p_child_node->parent=p_parent_node->parent; p_child_node->r_child=p_parent_node; p_parent_node->parent=p_child_node; }

插入和删除就比较容易了，主要是通过递归思想：

if(parent_node->key>p_node->key) { if(0==parent_node->l_child) { parent_node->l_child=p_node; p_node->parent=parent_node; if(head->l_child->key>p_node->key)//判断此时节点值是否比插入前的最小值更小，如果是的，那么让head 的左孩子节点指向现在这个节点 head->l_child=p_node; break; } else parent_node=parent_node->l_child; } else { if(0==parent_node->r_child) { parent_node->r_child=p_node; p_node->parent=parent_node; if(head->r_child->key<=p_node->key)//判断此时及诶但值是否比插入前的最大值更大或者相等，如果是的，那么让head 的右孩子节点指向现在这个节点 head->r_child=p_node; break; } else parent_node=parent_node->r_child; }

源码：

enum Color_node {red=0,black}; template<class T> struct node { typedef node* PVOID ; PVOID parent,l_child,r_child; T key; Color_node color; node(T value=T(0),Color_node c=Color_node::red):parent(0),l_child(0),r_child(0),key(value),color(c){} }; template<class T> class RB_Tree { public: typedef typename node<T>* PNODE; protected: PNODE head;//左孩子节点指向最小节点，右孩子节点指向最大节点，父节点指向跟节点 int number;//节点数目 public: RB_Tree():head(0),number(0) { init();//负责对head 进行构造 } protected://负责内存分配和初始化 PNODE create_node(T value) { PNODE new_node=(PNODE)calloc(1,sizeof(node<T>));//分配内存 new(new_node) node<T>(value);//palcement new 负责构造这段内存区域 return new_node; } void dealloc_node(PNODE pNode)//释放掉内存 { free(pNode); pNode=0; } void init() { head=create_node(T(0)); head->parent=0; head->l_child=head; head->r_child=head; head->color=Color_node::red; } private://负责执行插入，删除，维持平衡的实体 void l_rotaion(PNODE p_child_node,PNODE p_parent_node) { p_parent_node->r_child=p_child_node->l_child; if(p_child_node->l_child!=0) p_child_node->l_child->parent=p_parent_node; if(p_parent_node==head->parent) { head->parent=p_child_node; } else { if(p_parent_node==p_parent_node->parent->l_child) { p_parent_node->parent->l_child=p_child_node; } else { p_parent_node->parent->r_child=p_child_node; } } p_child_node->parent=p_parent_node->parent; p_child_node->l_child=p_parent_node; p_parent_node->parent=p_child_node; } void r_roation(PNODE p_child_node,PNODE p_parent_node) { p_parent_node->l_child=p_child_node->r_child; if(p_child_node->r_child!=0) p_child_node->r_child->parent=p_parent_node; if(p_parent_node==head->parent) { head->parent=p_child_node; } else { if(p_parent_node==p_parent_node->l_child) { p_parent_node->parent->l_child=p_child_node; } else { p_parent_node->parent->r_child=p_child_node; } } p_child_node->parent=p_parent_node->parent; p_child_node->r_child=p_parent_node; p_parent_node->parent=p_child_node; } void rebalance(PNODE root,PNODE p_node) { PNODE temp_node=p_node; if(0==root) return ; while(temp_node!=head->parent&&temp_node->parent->color==Color_node::red) { if((temp_node->parent->parent->r_child!=0)&&(temp_node->parent->parent->r_child->color==Color_node::red)&&(temp_node->parent->parent->l_child!=0)&&(temp_node->parent->parent->l_child->color==Color_node::red)) { temp_node->parent->color=Color_node::black; temp_node->parent->parent->color=Color_node::red; temp_node->parent->parent->r_child->color=Color_node::black; temp_node=temp_node->parent->parent; } else { if((temp_node==temp_node->parent->l_child)&&(temp_node->parent==temp_node->parent->parent->l_child)) { temp_node->parent->color=Color_node::black; temp_node->parent->parent->color=Color_node::red; this->r_roation(temp_node->parent,temp_node->parent->parent); } else { if((temp_node==temp_node->parent->r_child)&&(temp_node->parent==temp_node->parent->parent->l_child)) { this->l_rotaion(temp_node,temp_node->parent); temp_node->color=Color_node::black; temp_node->parent->color=Color_node::red; this->r_roation(temp_node,temp_node->parent); } else { if((temp_node==temp_node->parent->l_child)&&(temp_node->parent==temp_node->parent->parent->r_child)) { this->r_roation(temp_node,temp_node->parent); temp_node->color=Color_node::black; temp_node->parent->color=Color_node::red; this->l_rotaion(temp_node,temp_node->parent); } else { temp_node->parent->color=Color_node::black; temp_node->parent->parent->color=Color_node::red; this->l_rotaion(temp_node->parent,temp_node->parent->parent); } } } } } head->parent->color=Color_node::black; } void insert(PNODE root,PNODE p_node)//负责执行插入操作 { if(0==root)//首先判断跟节点是否为空 { head->parent=p_node; head->l_child=p_node; head->r_child=p_node; p_node->parent=head; root=p_node; } else { PNODE parent_node=root; while(1) { if(parent_node->key>p_node->key) { if(0==parent_node->l_child) { parent_node->l_child=p_node; p_node->parent=parent_node; if(head->l_child->key>p_node->key)//判断此时节点值是否比插入前的最小值更小，如果是的，那么让head 的左孩子节点指向现在这个节点 head->l_child=p_node; break; } else parent_node=parent_node->l_child; } else { if(0==parent_node->r_child) { parent_node->r_child=p_node; p_node->parent=parent_node; if(head->r_child->key<=p_node->key)//判断此时及诶但值是否比插入前的最大值更大或者相等，如果是的，那么让head 的右孩子节点指向现在这个节点 head->r_child=p_node; break; } else parent_node=parent_node->r_child; } } } this->rebalance(head->parent,p_node); } public://接口 T getMin() { return head->l_child->key; } T getMax() { return head->r_child->key; } void insert(T value) { PNODE new_node=create_node(value); insert(head->parent,new_node); } void earser(T value); };