如果在二叉查找树中一开始给定的数列是有序的,那么在构建二叉查找树时就会形成一条很长的链条式的树,此时对这棵树的查找复杂度将会变成O(n),起不到使用二叉查找树来进行数据查询优化的目的,于是需要对这颗树的结构进行调整,使树的高度在每次插入元素后仍然能保持O(logn)的级别,这样能让查询的时间仍然是O(logn),于是就产生了平衡二叉树。
平衡二叉树由前苏联两位数学家提出,因此一般也称为AVL树,AVL树仍然是一颗二叉查找树,只是在其基础上增加了“平衡”的要求。所谓平衡是指,对AVL树的任意结点来说,其左子树与右子树的高度之差的绝对值不超过1,其中左子树与右子树的高度之差称为该结点的平衡因子
只要能随时保证每个结点的平衡因子的绝对值不超过1, AVL的高度就始终能保持O(logn)级别,由于需要对每个结点都得到平衡因子,因此需要在树的结构中加入一个变量height,用来记录以当前结点为根结点的子树的高度。
struct node{
int v,height;//v为结点权值,height为当前子树高度
node *lchild,*rchild;//左右孩子结点地址
};
在这种定义下,如果需要新建一个结点,就可以采用如下写法:
//生成一个新结点,v为结点权值
node *newNode(int v){
node *Node=new node;//申请一个node型变量的地址
Node->v=v;
Node->h=1;//结点高度初始化为1
Node->l=NULL;//初始状态下没有左右孩子
Node->r=NULL;
return Node;//返回新建结点的地址
}
显然,可以通过下面的函数获取结点root所在子树的当前高度
//获取以root为根结点的子树的当前height
int getHeight(node *root){
if(root==NULL) return 0;//空结点的高度为0
return root->height;
}
于是根据定义,可以通过下面的函数来计算平衡因子
//计算结点root的平衡因子
int getBalanceFactor(node *root){
//左子树的高度-右子树的高度
return getHeight(root->lchild)-getHeight(root->rchild);
}
可以通过下面的函数来计算结点root的高度
//更新root的height
void updateHeight(node *root){
root->height=max(getHeight(root->lchild),getHeight(root->rchild))+1;
}