判断树是否为平衡二叉树

看穿数据之美

于 2017-11-14 11:09:39 发布

阅读量459

点赞数

分类专栏：数据结构算法

算法同时被 2 个专栏收录

51 篇文章

订阅专栏

数据结构

44 篇文章

订阅专栏

bool IsBalanced(BinaryTreeNode* pRoot, int* depth)  

{  

    if(pRoot== NULL)  

    {  

        *depth = 0;  

        return true;  

    }  

    int nLeftDepth,nRightDepth;  

    bool bLeft= IsBalanced(pRoot->m_pLeft, &nLeftDepth);  

    bool bRight = IsBalanced(pRoot->m_pRight, &nRightDepth);  

    if (bLeft && bRight)  

    {  

        int diff = nRightDepth-nLeftDepth;  

        if (diff<=1
 && diff>=-1)  

        {  

            *depth = 1+(nLeftDepth > nRightDepth ? nLeftDepth : nRightDepth);  

            return true;  

        }  

    }  

    return false;  

}  

bool IsBalanced(BinaryTreeNode* pRoot)  

{  

    int depth = 0;  

    return IsBalanced(pRoot, &depth);  

}