面试题 04.04. 检查平衡性

题

面试题 04.04. 检查平衡性
实现一个函数，检查二叉树是否平衡。在这个问题中，平衡树的定义如下：任意一个节点，其两棵子树的高度差不超过 1。

示例 1:
给定二叉树 [3,9,20,null,null,15,7]
3
/
9 20
/
15 7
返回 true 。
示例 2:
给定二叉树 [1,2,2,3,3,null,null,4,4]
1
/
2 2
/
3 3
/
4 4
返回 false 。

代码

class Solution
{
public:
    bool result = true;
    bool isBalanced(TreeNode *root)
    {
        dfs(root);
        return result;
    }

    int dfs(TreeNode *root)
    {
        if (!root || !result)
        {
            return 0;
        }

        int l = dfs(root->left);
        int r = dfs(root->right);

        if (abs(l-r)>1)
        {
            result = false;
            return 0;
        }

        if (l < r)
        {
            l = r;
        }
        return l + 1;
    }
};