欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

判断一颗二叉树是否为平衡二叉树

程序员文章站 2022-05-21 19:18:00
...

平衡二叉树:左子树和右子树的高度相差<=1,且左右子树都是平衡二叉树。
方法一:调用求二叉树高度的函数求每个节点的左右孩子深度,然后直接判断。
方法二:在方法一中,求该结点的的左右子树深度时遍历一遍树,再次判断子树的平衡性时又遍历一遍树结构,造成遍历多次。因此现在的做法是:一边遍历树一边判断每个结点是否具有平衡性。
具体代码看下面:

//二叉树的高度
int _Height(AVLTreeNode<K,V>* pRoot)
{
    if (pRoot == NULL)
        return 0;
    if (pRoot->_Left == NULL && pRoot->_Right == NULL)
        return 1;
    int heightL = _Height(pRoot->_Left);
    int heightR = _Height(pRoot->_Right);
    return (heightL > heightR) ? heightL + 1 : heightR + 1;
}
//(方法一)判断二叉树是否为平衡二叉树
bool _IsBalance(AVLTreeNode<K, V>* pRoot)
{
    if (NULL == pRoot)
        return true;
    int heightL = _Height(pRoot->_Left);
    int heightR = _Height(pRoot->_Right);
    if (abs(heightR - heightL) > 1)
        return false;
    return _IsBalance(pRoot->_Left) && _IsBalance(pRoot->_Right);
}
//(方法二)判断二叉树是否为平衡二叉树(优化版)
bool _IsBalance(AVLTreeNode<K, V>* pRoot,int *pDepth)
{
    if (NULL == pRoot)
    {
        *pDepth = 0;
        return true;
    }   
    int heightL, heightR;
    if (_IsBalance(pRoot->_Left, &heightL) && _IsBalance(pRoot->_Right,&heightR ))
    {
        if (abs(heightR - heightL) <= 1)
        {
            *pDepth = 1+(heightL > heightR ? heightL : heightR);
            return true;
        }   
    }
    return false;
}