算法导论红黑树的扩张动态顺序统计

最新推荐文章于 2020-07-07 00:04:28 发布

niewei1986

最新推荐文章于 2020-07-07 00:04:28 发布

阅读量430

点赞数

分类专栏：算法导论

本文链接：https://blog.csdn.net/niewei1986/article/details/72236272

版权

#include <stdio.h>    
#include <stdlib.h>    
#define RED 1    
#define BLACK 0    
    
typedef struct RBTreeNode    
{    
    int key;    
    int color; 
	int size;
    RBTreeNode *p;    
    RBTreeNode *left;    
    RBTreeNode *right;    
}RBT,*pRBT;    
    
pRBT nil=(pRBT)malloc(sizeof(RBT));    
int bh=0;  
  
void initNil()    
{    
    nil->key=-1;    
    nil->color=BLACK;    
    nil->p=NULL;    
    nil->left=NULL;    
    nil->right=NULL; 
	nil->size=0;
}    
    
void leftRotate(pRBT *root,pRBT x)    
{    
    //左旋要有右子树    
    if(x->right==nil)    
        return;    
    pRBT y=x->right;    
    x->right=y->left;    
    if(y->left != nil)    
        y->left->p=x;    
    y->p=x->p;    
    if(x->p==nil)    
    {    
        (*root)=y;    
    }    
    else if(x == x->p->left)    
    {    
        x->p->left=y;    
    }    
    else    
    {    
        x->p->right=y;    
    }    
    y->left=x;    
    x->p=y;   
	//维护子树数目
	y->size=x->size;
	x->size=x->left->size+x->right->size+1;
}    
    
void rightRotate(pRBT *root,pRBT x)    
{    
    //右旋要有左子树    
    if(x->left==nil)    
        return;    
    pRBT y=x->left;    
    x->left=y->right;
	if(y->right != nil)    
        y->right->p=x;
    y->p=x->p;    
    if(x->p==nil)    
    {    
        (*root)=y;    
    }    
    else if(x==x->p->left)    
    {    
        x->p->left=y;    
    }    
    else    
    {    
        x->p->right=y;    
    }    
    y->right=x;    
    x->p=y;   
	//维护子树数目
	x->size=y->size;
	y->size=y->left->size+y->right->size+1;
}    
    
void rbInsertFixup(pRBT *root,pRBT z)    
{    
    while(z->p->color=