9_2_2平衡二叉树.docVIP

/*************************平衡二叉树**********************/ typedef struct BSTNode { ElemType data; int bf; struct BSTNode *lchild,*rchild; } BSTNode, *BSTree void R_Rotate (BSTree p) { //以*p为根的二叉树右旋处理，旋转后p指向新的树根结点 lc = p-lchild; p-lchild = lc-rchild; lc-rchild = p; p = lc; } // R_Rotate void L_Rotate (BSTree p) { //以*p为根的二叉树左旋处理，旋转后p指向新的树根结点 rc=p-rchild; p-rchild=rc-lchild; rc-lchild=p; p=rc; } // R_Rotate #define LH 1 //左高 #define EH 0 //等高 #define RH -1 //右高 Status InsertAVL (BSTree T, ElemType e, int taller) { /*若在平衡的二叉排序树T中不存在和e相同关键字的结点，则插入，并返回1，否则返回0。若因插入而使二叉排序树失去平衡，则做相应的平衡旋转处理*，taller反映T长高与否*/ if(!T) //插入新结点，树长高 { T = (BSTree) malloc (sizeof(BSTNode)); T-data=e; T-lchild=T-rchild=NULL; T-bf=EH; taller=TRUE; } else { if (e==T-data) {taller=FALSE; return 0;} //已存在e，不插入 if (eT-data) //在左子树中搜索 { if (!InsertAVL(T-lchild,e,taller)) return 0;//在左子树中找到e不插入 if(taller) //已插入到左子树中 switch(T-bf) //检查原本*T的平衡度 { case LH: LeftBalance(T); taller=FALSE; break; case EH: T-bf=LH; taller=TRUE; break; case RH: T-bf=EH; taller=FALSE; break; } //switch } //if else { //在右子树中搜索 if( !InsertAVL(T-rchild,e,taller)) return 0; if( taller) switch(T-bf) { case LH: T-bf=EH; taller=FALSE; break; case EH: T-bf=RH; taller=TRUE; break; case RH: RightBalance(T); taller=FALSE; break; } //switch } //else } //else return 1; } //InsetAVL void LeftBalance (BSTree T) { lc=T-lchild; switch ( lc-bf ) //检查*T左子树平衡度 { case LH: //新结点插在*T的左孩子的左子树上 T-bf=lc-bf=EH; R_Rotate(T); break; case RH: //新结点插在*T的左孩子的右子树上 rd = lc-rchild; swith(rd-bf) { case LH: T-bf = RH; lc-bf = EH; break； case EH: T-bf = lc-bf = EH; break； case RH: T-bf = EH; lc-bf=LH; break； } //swith(rd-bf) rd-bf=EH; L_Rotate(T-lchild); //*T的左子树左旋 R_Rotate(T); //*T的右旋 }// switch(lc-bf) } //LeftBalance