AVL树

平衡二叉树,是一个方便查找的树,树的左子树深度与右子树的深度的差总(BF)是在+1,0,-1之中。

随着树的建立,插入,树都会自动的进行调整,使得其满足上面的条件。

1、+1表示左子树的深度比右子树的深度多1.

2、0 表示左子树的深度与右子树的深度相同。

3、-1表示左子树的深度比右子树神的小1.

因此,如果一个数据插入到情况1中,也就是说,数据插入到左子树中,左子树的深度将会比右子树多2.此时,需要调整树的结构。如果插入尾端节点的左子树中,则这个尾端节点相应的BF值,就变成+1.相反,如果插入到它的右子树中,BF值就会变成-1.这个调整也会返回到上面一层的节点,再次进行调整。

 

这里相应的介绍一个左旋,与右旋的基本知识。

比如下图

在进行左旋时,将会发生下面的情况:

void L_Rotate(BiTree *p){
    //传入根节点进行右旋
    BiTree R;
    R = (*p)->rchild;
    (*p)->rchild = R->lchild;
    R->lchild = (*p);
    (*p) = R;
}

最后子树将会变成

 

相应的右旋,则运行下面的代码

void R_Rotate(BiTree *p){
    //传入一个根节点,进行右旋。定义它的左子树节点为L,根节点的左子树变成L的右子树,L的右子树变成根节点。最后把根节点指针指向L
    BiTree L;
    L = (*p)->lchild;
    (*p)->lchild =  L->rchild;
    L->rchild = (*p);
    (*p) = L;
}

 

了解左旋与右旋后,就该进行树的调整介绍了。

这里有一个技巧:

1 如果插入的元素插入到左子树,使得左子树的BF值发生改变。如果左子树节点的BF值,与根节点的BF值相同符号,则进行一次右旋,即可。但是如果是不同符号,则要进行双旋(即先进性左旋,使得子树高度加一,在进行右旋,平衡子树)

2 如果插入到右子树,也观察符号,相同,则进行一次右旋,如果不同,则进行双旋。

代码如下

void LeftBalance(BiTree *T){
    BiTree L,Lr;
    L = (*T)->lchild;
    switch(L->bf){
    case LH://符号与根相同,因此进行右旋一次,就行了
        (*T)->bf = L->bf = EH;
        R_Rotate(T);
        break;
    case RH://符号与根不同,要进行双旋;对子树进行一次旋转,再对T进行一次旋转
        Lr = L->rchild;
        switch(Lr->bf){
        case LH: //如果是左子数高,那么对根节点赋值为-1,因为没有右子树,根节点将会出现左子树为空的情况;左子树旋转后,会平衡为EH
            (*T)->bf = RH;
            L->bf = EH;
            break;
        case EH://如果为平衡,那么根节点和左子树节点都为平衡EH
            (*T)->bf = L->bf = EH;
            break;
        case RH://如果右子树高,那么对根节点赋值为0,对左子树赋值为+1,因为进行旋转后,左子树节点的右子树会空。根节点EH
            (*T)->bf = EH;
            L->bf = LH;
            break;
        }
        Lr->bf = EH;
        L_Rotate(&(*T)->lchild);
        R_Rotate(T);
    }
}

void RightBalance(BiTree *T){
    BiTree R,Rl;
    R = (*T)->rchild;
    switch(R->bf){
    case RH:
        (*T)->bf = R->bf = EH;
        L_Rotate(T);
        break;
    case LH:
        Rl = R->lchild;
        switch(Rl->bf){
        case LH: //如果是左子数高,那么对根节点赋值为-1,因为没有右子树,根节点将会出现左子树为空的情况;左子树旋转后,会平衡为EH
            (*T)->bf = EH;
            R->bf = RH;
            break;
        case EH://如果为平衡,那么根节点和左子树节点都为平衡EH
            (*T)->bf = R->bf = EH;
            break;
        case RH://如果右子树高,那么对根节点赋值为0,对左子树赋值为+1,因为进行旋转后,左子树节点的右子树会空。根节点EH
            (*T)->bf = LH;
            R->bf = EH;
            break;
        }
        Rl->bf = EH;
        R_Rotate(&(*T)->rchild);
        L_Rotate(T);
    }
}

插入时,要遍历到子树的最底层,进行分析,逐层的改变BF值,进行平衡。知道标记taller为0时,表示对深度不发生改变,就不需要向上遍历了。

int insertAVL(BiTree *T,int e, int *taller){
    if(!(*T)){
        (*T)=(BiTree)malloc(sizeof(BiTNode));
        (*T)->data = e;
        (*T)->lchild = NULL;
        (*T)->rchild = NULL;
        (*T)->bf = EH;
        *taller = 1;
    }else{
        if(e == (*T)->data){ //存在相同节点
            *taller = 0;
            return 0;
        }
        if(e < (*T)->data){
            if(!insertAVL(&(*T)->lchild,e,taller))
                return 0;
            if(taller){
                switch((*T)->bf){
                case LH:
                    LeftBalance(T);
                    *taller = 0;
                    break;
                case EH:
                    (*T)->bf = LH;
                    *taller = 1;
                    break;
                case RH:
                    (*T)->bf = EH;
                    *taller = 0;
                    break;
                }
            }
        }else{
            if(!insertAVL(&(*T)->rchild,e,taller))
                return 0;
            if(taller){
                switch((*T)->bf){
                case LH:
                    (*T)->bf = 0;
                    *taller = 0;
                    break;
                case EH:
                    (*T)->bf=RH;
                    *taller = 1;
                    break;
                case RH:
                    RightBalance(T);
                    *taller = 0;
                    break;
                }
            }
        }
    }
    return 1;
}

全部代码:

#include <stdio.h>
#include <stdlib.h>
#define LH +1
#define EH 0
#define RH -1

typedef struct BiTNode{
    int data;
    int bf;
    struct BiTNode *lchild,*rchild;
}BiTNode,*BiTree;

void R_Rotate(BiTree *p);
void L_Rotate(BiTree *p);
void LeftBalance(BiTree *p);
void RightBalance(BiTree *T);
int insertAVL(BiTree *T,int e, int *taller);
void InOrderTree(BiTree b);

int main(){
    int i;
    int a[10]={4,7,9,1,2,3,0,5,6,8};
    BiTree T = NULL;
    int *taller = (int *)malloc(sizeof(int));
    for(i=0;i<10;i++){
        insertAVL(&T,a[i],taller);
        InOrderTree(T);
        printf("\n");
    }
    getchar();
}
void InOrderTree(BiTree b){
    if( b== NULL)
        return;
    InOrderTree(b->lchild);
    printf("%d ",b->data);
    InOrderTree(b->rchild);
}
int insertAVL(BiTree *T,int e, int *taller){
    if(!(*T)){
        (*T)=(BiTree)malloc(sizeof(BiTNode));
        (*T)->data = e;
        (*T)->lchild = NULL;
        (*T)->rchild = NULL;
        (*T)->bf = EH;
        *taller = 1;
    }else{
        if(e == (*T)->data){ //存在相同节点
            *taller = 0;
            return 0;
        }
        if(e < (*T)->data){
            if(!insertAVL(&(*T)->lchild,e,taller))
                return 0;
            if(taller){
                switch((*T)->bf){
                case LH:
                    LeftBalance(T);
                    *taller = 0;
                    break;
                case EH:
                    (*T)->bf = LH;
                    *taller = 1;
                    break;
                case RH:
                    (*T)->bf = EH;
                    *taller = 0;
                    break;
                }
            }
        }else{
            if(!insertAVL(&(*T)->rchild,e,taller))
                return 0;
            if(taller){
                switch((*T)->bf){
                case LH:
                    (*T)->bf = 0;
                    *taller = 0;
                    break;
                case EH:
                    (*T)->bf=RH;
                    *taller = 1;
                    break;
                case RH:
                    RightBalance(T);
                    *taller = 0;
                    break;
                }
            }
        }
    }
    return 1;
}

void R_Rotate(BiTree *p){
    //传入一个根节点,进行右旋。定义它的左子树节点为L,根节点的左子树变成L的右子树,L的右子树变成根节点。最后把根节点指针指向L
    BiTree L;
    L = (*p)->lchild;
    (*p)->lchild =  L->rchild;
    L->rchild = (*p);
    (*p) = L;
}
void L_Rotate(BiTree *p){
    //传入根节点进行右旋
    BiTree R;
    R = (*p)->rchild;
    (*p)->rchild = R->lchild;
    R->lchild = (*p);
    (*p) = R;
}
void LeftBalance(BiTree *T){
    BiTree L,Lr;
    L = (*T)->lchild;
    switch(L->bf){
    case LH://符号与根相同,因此进行右旋一次,就行了
        (*T)->bf = L->bf = EH;
        R_Rotate(T);
        break;
    case RH://符号与根不同,要进行双旋;对子树进行一次旋转,再对T进行一次旋转
        Lr = L->rchild;
        switch(Lr->bf){
        case LH: //如果是左子数高,那么对根节点赋值为-1,因为没有右子树,根节点将会出现左子树为空的情况;左子树旋转后,会平衡为EH
            (*T)->bf = RH;
            L->bf = EH;
            break;
        case EH://如果为平衡,那么根节点和左子树节点都为平衡EH
            (*T)->bf = L->bf = EH;
            break;
        case RH://如果右子树高,那么对根节点赋值为0,对左子树赋值为+1,因为进行旋转后,左子树节点的右子树会空。根节点EH
            (*T)->bf = EH;
            L->bf = LH;
            break;
        }
        Lr->bf = EH;
        L_Rotate(&(*T)->lchild);
        R_Rotate(T);
    }
}

void RightBalance(BiTree *T){
    BiTree R,Rl;
    R = (*T)->rchild;
    switch(R->bf){
    case RH:
        (*T)->bf = R->bf = EH;
        L_Rotate(T);
        break;
    case LH:
        Rl = R->lchild;
        switch(Rl->bf){
        case LH: //如果是左子数高,那么对根节点赋值为-1,因为没有右子树,根节点将会出现左子树为空的情况;左子树旋转后,会平衡为EH
            (*T)->bf = EH;
            R->bf = RH;
            break;
        case EH://如果为平衡,那么根节点和左子树节点都为平衡EH
            (*T)->bf = R->bf = EH;
            break;
        case RH://如果右子树高,那么对根节点赋值为0,对左子树赋值为+1,因为进行旋转后,左子树节点的右子树会空。根节点EH
            (*T)->bf = LH;
            R->bf = EH;
            break;
        }
        Rl->bf = EH;
        R_Rotate(&(*T)->rchild);
        L_Rotate(T);
    }
}
View Code

运行实例:

 

 

posted @ 2014-04-21 17:02  xingoo  阅读(998)  评论(0编辑  收藏  举报