红黑树系列之旋转C++实现

(1)概述 

         二叉树是使用非常广泛的数据结构，但如果是常规的插入，会导致二叉树的高度过高和出现整棵树不平衡的情况。红黑树是一种平衡二叉树，C++STL中的set,map及其扩展容器内部的数据结构都是红黑树。

(2)左旋转

         比如说，需要把x旋转为y的左结点。整个算法的思路非常清晰：从上至下，先得到y指针，讲x的右指针指向y的左结点，然后利用parent函数得到x的父亲结点，如果为NULL，则y为新的根，如果不为NULL，则根据x是其父亲的左孩子还是右孩子，将指针指向y。最后将y的左指针指向x，完成旋转。值得注意的是，算法是具有顺序的逻辑步骤，不能够调换顺序，如果改变赋值的顺序会造成内存失去指针指向，出现内存错误。

       代码：注：parent为求父亲结点的函数，root是始终指向根结点内存区域的指针。

[cpp] view plain copy

1. //左旋转，假设x->pRight!=NULL  
2. void left_rotate(NODE *head,NODE *x)//head是根结点，x是待左旋转的结点  
3. {  
4.     if(x->pRight!=NULL)  
5.     {  
6.         NODE *y=x->pRight;  
7.         if(y->pLeft!=NULL)  
8.             x->pRight=y->pLeft;  
9.         NODE *px=parent(x,head);  
10.         if(px==NULL)//如果x是根结点，那么就把y置为根结点  
11.             root=y;  
12.         else if(px->pLeft==x)  
13.             px->pLeft=y;  
14.         else  
15.             px->pRight=y;  
16.         y->pLeft=x;  
17.     }  
18.     else  
19.         printf("item为%ld的结点不能够进行左旋转！",x->item);  
20. }  

(3)右旋转

        方法与左旋转基本相同，只是方向相反，不再赘述其过程。

        代码：

[cpp] view plain copy

1. //右旋转，假设y->pLeft!=NULL     
2. void right_rotate(NODE *head,NODE *y)//head是根结点，y是待右旋转的结点  
3. {  
4.     if(y->pLeft!=NULL)  
5.     {  
6.         NODE *x=y->pLeft;  
7.         if(x->pRight!=NULL)  
8.             y->pLeft=x->pRight;  
9.         NODE *py=parent(y,head);  
10.         if(py==NULL)  
11.             root=x;  
12.         else if(py->pLeft==y)  
13.             py->pLeft=x;  
14.         else  
15.             py->pRight=x;  
16.         x->pRight=y;  
17.     }  
18.     else  
19.         printf("item为%ld的结点不能够进行右旋转！",y->item);  
20. }  

[cpp] view plain copy

1. //返回父亲结点  
2. NODE *parent(NODE *pNode,NODE *head)  
3. {  
4.     NODE *result=NULL;  
5.     if(head!=NULL)  
6.     {  
7.         if(head->pLeft==pNode || head->pRight==pNode)  
8.             return head;  
9.         if(head->pLeft!=NULL)  
10.         {  
11.             result=parent(pNode,head->pLeft);  
12.             if(result!=NULL)//找到之后就不搜索其他的了  
13.                 return result;  
14.         }  
15.         if(head->pRight!=NULL)  
16.         {  
17.             result=parent(pNode,head->pRight);  
18.             if(result!=NULL)  
19.                 return result;  
20.         }  
21.     }  
22.     return result;//没有找到，返回NULL  
23. }  

        总结：旋转的算法思路非常清晰，整个逻辑思考是重点。

[转载自 http://blog.csdn.net/neilhappy/article/details/7209873]