rbTree.h
#ifndef RBTREE_H_INCLUDED #define RBTREE_H_INCLUDED #undef NULL #if defined(__cplusplus) #define NULL 0 #else #define NULL ((void *)0) #endif /* 红黑树是二叉查找树的一种且具有以下性质: 1:每个节点要么是红色要么是黑色 2:根节点和叶子节点都是黑色的 3:如果一个结点是红的,那么它的两个儿子都是黑的 4:每一条路径上黑节点的数目都相同 */ /* 左旋: right是node的右孩子。-----条件 旋转之后node成为right的左孩子。 right的左孩子成为node的右孩子。 node的左孩子不变,right的右孩子不变。 node right / \ ==> / \ a right node y / \ / \ b y a b */ /* 右旋: node left / \ / \ left y ==> a node / \ / \ a b b y 父节点的左孩子才能右旋 右旋后父节点和左子树关系交换。 父节点变成左孩子的右孩子 左孩子的左孩子(a)位置不变 父节点的右孩子(y)位置不变 */ typedef enum en_color { RED = 0, BLK }COLOR; typedef struct tag_rb_t { struct tag_rb_t *pstLt; //左孩子节点 struct tag_rb_t *pstRt; //右孩子节点 struct tag_rb_t *pstPt; //双亲节点 COLOR color; //节点的颜色 int key; //key }rb_t; //左旋 void rb_LeftRotate(rb_t* pNode, rb_t** ppRoot); //右旋 void rb_RightRotate(rb_t* pNode, rb_t** ppRoot); //插入时修正左子树 void rb_InsertFixupLeft(rb_t *pGrand, rb_t **pParent, rb_t *pUncle, rb_t **ppNode, rb_t **ppRoot); //插入时修正右子树 void rb_InsertFixupRight(rb_t *pGrand, rb_t **pParent, rb_t *pUncle, rb_t **ppNode, rb_t **ppRoot); //插入时修正左右子树的入口 void rb_InsertFixup(rb_t *pNode, rb_t **ppRoot); //插入 int rb_Insert(rb_t *pNode, rb_t **ppRoot); //删除节点 void rb_Delete(rb_t *pDel, rb_t **ppRoot); //查找一个节点 rb_t* rb_Search(int key); #endif // RBTREE_H_INCLUDED
rbTree.c
#include "stdafx.h" #include "rbTree.h" #define RETURN_ERR -1 #define RETURN_OK 0 /***************************************************************************** 函数功能:红黑树插入节点时左旋 函数入参:rb_t* pNode 左旋操作前的父节点 函数出参:rb_t** ppRoot 红黑树的根节点,在左旋过程中有可能需要改变根节点的位置 函数返回值:无 其他:左旋操作参看头文件中的说明 ******************************************************************************/ void rb_LeftRotate(rb_t* pNode, rb_t** ppRoot) { //pNode的右孩子。左旋发生在父节点(pNode)和右孩子(ppstRt)之间。左旋完成之后ppstRt成为父节点,pNode成为左孩子 rb_t *ppstRt = pNode->pstRt; //ppstRt的左孩子需要成为pNode的右孩子,pNode的左孩子不变,ppstRt的右孩子不变。 if (NULL != (pNode->pstRt = ppstRt->pstLt)) { ppstRt->pstLt->pstPt = pNode; //ppstRt的左孩子成为node的右孩子,所以ppstRt的左孩子的父节点需要由ppstRt修改成pNode } ppstRt->pstLt = pNode; //交换pNode和ppstRt的父子关系。 if (NULL != (ppstRt->pstPt = pNode->pstPt)) //修改父节点。如果不为空说明pNode不是红黑树的根节点 { //左旋后,未变动的节点的父节点也需要修改 if (pNode == pNode->pstPt->pstRt) { pNode->pstPt->pstRt = ppstRt; } else { pNode->pstPt->pstLt = ppstRt; } } else { *ppRoot = ppstRt; //pNode是红黑树的根节点,此时根节点需要修改成为左旋前pNode的右孩子 } pNode->pstPt = ppstRt; return; } /***************************************************************************** 函数功能:红黑树插入节点时右旋 函数入参:rb_t* pNode 右旋前的父节点 函数出参:rb_t** ppRoot 红黑树的根节点 函数返回值:无 其他:右旋操作参看头文件中的说明 ******************************************************************************/ void rb_RightRotate(rb_t* pNode, rb_t** ppRoot) { rb_t *ppstLt = pNode->pstLt; if (NULL != (pNode->pstLt = ppstLt->pstRt)) { ppstLt->pstRt->pstRt = pNode; } ppstLt->pstRt = pNode; if (NULL != (ppstLt->pstPt = pNode->pstPt)) { if (pNode == pNode->pstPt->pstRt) { pNode->pstPt->pstRt = ppstLt; } else { pNode->pstPt->pstLt = ppstLt; } } else { *ppRoot = ppstLt; } pNode->pstPt = ppstLt; return; } /***************************************************************************** 函数功能:红黑树插入节点时调整左子树 函数入参:rb_t *pGrand, 祖父节点 rb_t **ppstPt, 父节点 rb_t* *pUncle, 父节点的兄弟节点 rb_t **ppNode, 待调整的节点 rb_t **ppRoot 红黑树的根节点 函数出参:无 函数返回值:无 其他: ******************************************************************************/ void rb_InsertFixupLeft(rb_t *pGrand, rb_t **pppstPt, rb_t *pUncle, rb_t **ppNode, rb_t **ppRoot) { rb_t *pNodeTmp = NULL; bool IsTrue = false; IsTrue = ((NULL != pUncle) && (RED == pUncle->color)); //uncle存在且是红色 if (IsTrue) { pUncle->color = BLK; (*pppstPt)->color = BLK; pGrand->color = RED; (*ppNode) = pGrand; //将祖父当做新增结点z,指针z上移俩层,且着为红色。 return; } //uncle不存在,或者是黑色的 if ((*ppNode) == (*pppstPt)->pstRt) //pNode是右孩子,左旋的条件 { rb_LeftRotate(*pppstPt, ppRoot); pNodeTmp = *pppstPt; *pppstPt = *ppNode; *ppNode = pNodeTmp; } //uncle是黑色的,此时pNode成为了左孩子。 (*pppstPt)->color = BLK; pGrand->color = RED; rb_RightRotate(pGrand, ppRoot); return; } /***************************************************************************** 函数功能:红黑树插入节点时调整右子树 函数入参:rb_t *pGrand, 祖父节点 rb_t **ppstPt, 父节点 rb_t* *pUncle, 父节点的兄弟节点 rb_t **ppNode, 待调整的节点 rb_t **ppRoot 红黑树的根节点 函数出参:无 函数返回值:无 其他: ******************************************************************************/ void rb_InsertFixupRight(rb_t *pGrand, rb_t **pppstPt, rb_t *pUncle, rb_t **ppNode, rb_t **ppRoot) { rb_t *pNodeTmp = NULL; bool IsTrue; IsTrue = ((NULL != pUncle) && (RED == pUncle->color)); if (IsTrue) { pUncle->color = BLK; (*pppstPt)->color = BLK; pGrand->color = RED; (*ppNode) = pGrand; return; } if ((*ppNode) == (*pppstPt)->pstLt) { rb_RightRotate(*pppstPt, ppRoot); pNodeTmp = *pppstPt; *pppstPt = *ppNode; *ppNode = pNodeTmp; } (*pppstPt)->color = BLK; pGrand->color = RED; rb_LeftRotate(pGrand, ppRoot); return; } /***************************************************************************** 函数功能:红黑树插入节点时调整子树 函数入参:rb_t *pNode, 带插入节点 rb_t **ppRoot 红黑树的根节点 函数出参:无 函数返回值:无 其他: ******************************************************************************/ void rb_InsertFixup(rb_t *pNode, rb_t **ppRoot) { rb_t *ppstPt = NULL; rb_t *pGrand = NULL; rb_t *pUncle = NULL; while ((NULL != (ppstPt = pNode->pstPt)) && (RED == ppstPt->color)) { //ppstPt是pNode的父节点,且父节点是红色的 pGrand = ppstPt->pstPt; if (ppstPt == pGrand->pstLt) { pUncle = pGrand->pstRt; rb_InsertFixupLeft(pGrand, &ppstPt, pUncle, &pNode, ppRoot); } else { pUncle = pGrand->pstLt; rb_InsertFixupRight(pGrand, &ppstPt, pUncle, &pNode, ppRoot); } } (*ppRoot)->color = BLK; return; } /***************************************************************************** 函数功能:红黑树插入节点 函数入参:rb_t *pNode, 插入节点 rb_t **ppRoot 红黑树的根节点 函数出参:无 函数返回值:无 其他: ******************************************************************************/ int rb_Insert(rb_t *pNode, rb_t **ppRoot) { rb_t **ppNodeTmp = ppRoot; rb_t *ppstPt = NULL; //二叉查找树插入方法相同 while (NULL != (*ppRoot)) { ppstPt = *ppNodeTmp; if (pNode->key > (*ppNodeTmp)->key) { ppNodeTmp = &((*ppNodeTmp)->pstRt); } else if (pNode->key < (*ppNodeTmp)->key) { ppNodeTmp = &((*ppNodeTmp)->pstLt); } else { return RETURN_ERR; } } *ppNodeTmp = pNode; pNode->pstPt = ppstPt; pNode->color = RED; pNode->pstLt = NULL; pNode->pstRt = NULL; rb_InsertFixup(pNode, ppRoot); return RETURN_OK; } /***************************************************************************** 函数功能:根据key,查找节点 函数入参:int key rb_t *pRoot 红黑树的根节点 函数出参:无 函数返回值:key对应的节点 其他: ******************************************************************************/ rb_t* rb_Search(int key, rb_t *pRoot) { rb_t *pTmp = pRoot; while (NULL != pTmp) { if (pTmp->key < key) { pTmp = pTmp->pstLt; } else if (pTmp->key > key) { pTmp = pTmp->pstRt; } else { return pTmp; } } return NULL; } void rb_DeletepDelHaveTwoChildren(rb_t *pDel, rb_t **ppRoot, COLOR *peColor, rb_t **ppDelNxtChild, rb_t **ppDelNxtParent) { rb_t *pTmp = pDel; rb_t *pDelNext = NULL; //待删除节点的后继节点 rb_t *pDelNxtChild = NULL; //后继节点的右孩子 rb_t *pDelNxtParent = NULL; //后继节点的父节点 //查找pDel的后继 pDelNext = pDel->pstRt; pDelNext = pDelNext->pstLt; while (NULL != pDelNext) { pDelNext = pDelNext->pstLt; } pDelNxtChild = pDelNext->pstRt; pDelNxtParent = pDelNext->pstPt; *peColor = pDelNext->color; //后续节点存在右孩子 if (NULL != pDelNxtChild) { pDelNxtChild->pstPt = pDelNxtParent; } if (NULL != pDelNxtParent) { if (pDelNxtParent->pstLt == pDelNext) { pDelNxtParent->pstLt = pDelNxtChild; } else { pDelNxtParent->pstRt = pDelNxtChild; } } else //后续节点为空,说明待删除的节点时根节点,需要修改根节点 { *ppRoot = pDelNxtChild; } if (pDelNext->pstPt == pTmp) { pDelNxtParent = pDelNext; } pDelNext->pstPt = pTmp->pstPt; pDelNext->color = pTmp->color; pDelNext->pstRt = pTmp->pstRt; pDelNext->pstLt = pTmp->pstLt; if (pTmp->pstPt) { if (pTmp->pstPt->pstLt == pTmp) { pTmp->pstPt->pstLt = pDelNext; } else { pTmp->pstPt->pstRt = pDelNext; } } else { *ppRoot = pDel; } pTmp->pstLt->pstPt = pDel; if (pTmp->pstRt) { pTmp->pstRt->pstPt = pDel; } *ppDelNxtChild = pDelNxtChild; *ppDelNxtParent = pDelNxtParent; return; } void rb_DeletepDelNoTwoChild(rb_t *pDel, rb_t **ppRoot, COLOR *peColor, rb_t **ppDelNxtChild, rb_t **ppDelNxtParent) { rb_t *pTmp = pDel; rb_t *pDelNext = NULL; //待删除节点的后继节点 rb_t *pDelNxtChild = NULL; //后继节点的孩子 rb_t *pDelNxtParent = NULL; //后继节点的父节点 //只有一个孩子的情况 if (NULL != pDel->pstLt) { pDelNxtChild = pDel->pstRt; } else if (NULL != pDel->pstRt) { pDelNxtChild = pDel->pstLt; } pDelNxtParent = pDel->pstPt; *peColor = pDel->color; //修改待删除节点的孩子节点的父节点 if (pDelNxtChild) { pDelNxtChild->pstPt = pDelNxtParent; } if (pDelNxtParent) { if (pDelNxtParent->pstLt == pDel) { pDelNxtParent->pstLt = pDelNxtChild; } else { pDelNxtParent->pstRt = pDelNxtChild; } } else //父节点为空说明待删除的节点是根节点,需要修改根节点 { *ppRoot = pDelNxtChild; } *ppDelNxtChild = pDelNxtChild; *ppDelNxtParent = pDelNxtParent; return; } void rb_DeleteNode(rb_t *pDel, rb_t **ppRoot, COLOR *peColor, rb_t **ppDelNxtChild, rb_t **ppDelNxtParent) { //待删除的节点既有左孩子又有右孩子 if ((NULL != pDel->pstLt) && (NULL != pDel->pstRt)) { rb_DeletepDelHaveTwoChildren(pDel, ppRoot, peColor, ppDelNxtChild, ppDelNxtParent); } else { rb_DeletepDelNoTwoChild(pDel, ppRoot, peColor, ppDelNxtChild, ppDelNxtParent); } return; } void rb_DelFixupLeft(rb_t **ppDelNxtChild, rb_t *pDelNxtParent, rb_t **ppRoot) { rb_t *pTmp; rb_t *pTmpLeft; pTmp = pDelNxtParent->pstRt; if (pTmp->color == RED) //情况1:待删除节点的兄弟pTmp是红色的 { pTmp->color = BLK; pDelNxtParent->color = RED; //上俩行,改变颜色,pTmp->黑、待删除的节点的父节点->红。 rb_LeftRotate(pDelNxtParent, ppRoot); //再对待删除的节点的父节点做一次左旋 pTmp = pDelNxtParent->pstRt; //待删除节点的新兄弟new w 是旋转之前w的某个孩子。其实就是左旋后的效果。 } if ((!pTmp->pstLt || pTmp->pstLt->color == BLK) && (!pTmp->pstRt || pTmp->pstRt->color == BLK)) //情况2:x的兄弟w是黑色,且w的俩个孩子也都是黑色的 { //由于w和w的俩个孩子都是黑色的,则在x和w上得去掉一黑色, pTmp->color = RED; //于是,兄弟w变为红色。 *ppDelNxtChild = pDelNxtParent; //p[x]为新结点x pDelNxtParent = (*ppDelNxtChild)->pstPt; //x<-p[x] } else //情况3:x的兄弟w是黑色的, 且,w的左孩子是红色,右孩子为黑色。 { if (!pTmp->pstRt || pTmp->pstRt->color == BLK) { if ((pTmpLeft = pTmp->pstLt)) //w和其左孩子pstLt[w],颜色交换。 { pTmpLeft->color = BLK; //w的左孩子变为由黑->红色 } pTmp->color = RED; //w由黑->红 rb_RightRotate(pTmp, ppRoot); //再对w进行右旋,从而红黑性质恢复。 pTmp = pDelNxtParent->pstRt; //变化后的,父结点的右孩子,作为新的兄弟结点 } //情况4:x的兄弟w是黑色的 pTmp->color = pDelNxtParent->color; //把兄弟节点染成当前节点父节点的颜色。 pDelNxtParent->color = BLK; //把当前节点父节点染成黑色 if (pTmp->pstRt) //且w的右孩子是红 { pTmp->pstRt->color = BLK; //兄弟节点w右孩子染成黑色 } rb_LeftRotate(pDelNxtParent, ppRoot); //并再做一次左旋 *ppDelNxtChild = *ppRoot; //并把x置为根。 return; } return; } void rb_DelFixupRight(rb_t **ppDelNxtChild, rb_t *pDelNxtParent, rb_t **ppRoot) { rb_t *pTmp; rb_t *pTmpRight; pTmp = pDelNxtParent->pstLt; if (pTmp->color == RED) { pTmp->color = BLK; pDelNxtParent->color = RED; rb_RightRotate(pDelNxtParent, ppRoot); pTmp = pDelNxtParent->pstLt; } if ((!pTmp->pstLt || pTmp->pstLt->color == BLK) && (!pTmp->pstRt || pTmp->pstRt->color == BLK)) { pTmp->color = RED; *ppDelNxtChild = pDelNxtParent; pDelNxtParent = (*ppDelNxtChild)->pstPt; } else { if (!pTmp->pstLt || pTmp->pstLt->color == BLK) { if ((pTmpRight = pTmp->pstRt)) { pTmpRight->color = BLK; } pTmp->color = RED; rb_LeftRotate(pTmp, ppRoot); pTmp = pDelNxtParent->pstLt; } pTmp->color = pDelNxtParent->color; pDelNxtParent->color = BLK; if (pTmp->pstLt) { pTmp->pstLt->color = BLK; } rb_RightRotate(pDelNxtParent, ppRoot); *ppDelNxtChild = *ppRoot; return; } return; } void rb_DelFixup(rb_t *pDelNxtChild, rb_t *pDelNxtParent, rb_t **ppRoot) { while ((!pDelNxtChild || pDelNxtChild->color == BLK) && pDelNxtChild != *ppRoot) { if (pDelNxtParent->pstLt == pDelNxtChild) { rb_DelFixupLeft(&pDelNxtChild, pDelNxtParent, ppRoot); } else { rb_DelFixupRight(&pDelNxtChild, pDelNxtParent, ppRoot); } } if (pDelNxtChild) { pDelNxtChild->color = BLK; } return; } /***************************************************************************** 函数功能:将一个节点从红黑树中删除(只是将节点从红黑树中摘掉,节点的内存不会再本函数中释放) 函数入参:rb_t *pDel rb_t **ppRoot 函数出参:无 函数返回值:无 特别说明:pDel的内存需要在调用此函数之后,手动释放 ******************************************************************************/ void rb_Delete(rb_t *pDel, rb_t **ppRoot) { COLOR color; rb_t *pDelNxtChild = NULL; rb_t *pDelNxtParent = NULL; //将待删除的节点从红黑树中摘掉 rb_DeleteNode(pDel, ppRoot, &color, &pDelNxtChild, &pDelNxtParent); if (color == BLK) { rb_DelFixup(pDelNxtChild, pDelNxtParent, ppRoot); //调用rb_erase_rebalance来恢复红黑树性质 } return; }
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。