平衡二叉树

最新推荐文章于 2021-06-17 10:53:20 发布

freedom098

最新推荐文章于 2021-06-17 10:53:20 发布

阅读量159

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/freedom098/article/details/79137131

版权

leetcode 专栏收录该内容

129 篇文章 0 订阅

订阅专栏

输入一棵二叉树，判断该二叉树是否是平衡二叉树。

深搜，每次带着一个深度信息搜索，深搜的返回值就是是否平衡。
每次遍历到某一个节点的时候，先判断之前子树是不是平衡，如果平衡，深度自增，交到上一层去比较是否满足平衡条件。
递归的出口条件是，遍历到空值时，说明已经到叶子节点的下一层，深度值设置为0，同时返回true，这是返回true的原因和计算二叉树深度的原因类似。
另外深度增加的时候，只有在满足平衡条件时增加即可，因为只要有其中某一节点不平衡，其实后面深度也不用增加了，整个结果必定为false。

bool IsBalanced(TreeNode* pRoot, int &depth)
{
    if(pRoot == NULL)
    {
        depth = 0;
        return true;
    }

    int left,right;
    if(IsBalanced(pRoot->left, left) && IsBalanced(pRoot->right, right))
    {
        int diff = left-right;
        if(diff <= 1 && diff >= -1)
        {
            depth = 1+(left>right ? left:right);
            return true;
        }
    }

    return false;

}

bool IsBalanced_Solution(TreeNode* pRoot)
{
    int depth = 0;
    return IsBalanced(pRoot, depth);
}