考研数据结构(每日一题)
题目:判断二叉树是否是平衡二叉树的算法。
算法思想:
设二叉树的平衡标记为balance,以标记返回二叉树bt是否为平衡二叉树,若为平衡二叉树,则返回1,否则返回0,h为二叉树bt 的高度,采用后序遍历递归算法。
若bt为空,则高度为0,balance=1
若bt仅有根结点,则高度为1,balanc=1
否则,对bt的左右子树进行递归,返回左右子树的高度和平衡标记,bt的高度为最高子树的高度加1
若左右子树的高度差大于1,则balance=0
若左右子树的高度差小于等于1,且左右子树都平衡时,balance=1,否则balance=0
完整代码:
void Judge_AVL(Bitree bt,int &balance,int &h){
int bl = 0,br = 0,hl = 0,hr = 0; //设置左右子树的平衡标记和高度
if (bt == NULL) //空树,高度为0
{
h = 0;
balance = 1;
}
else if(bt -> lchild == NULL && bt -> rchild == NULL){
//仅有根结点,则高度为1
h = 1;
balance = 1;
}
else{
Judge_AVL(bt -> lchild,bl,hl); //递归判断左子树
Judge_AVL(bt -> rchild,br,hr); //递归判断右子树
h = (hl > hr ? hr: hr) + 1;
if(abs(hl - hr) < 2){ //若子树高度差的绝对值小于2,则看左右子树是否都平衡
balance = bl && br; //左右子树都平衡,二叉树平衡
}else{
balance = 0;
}
}
}