面试题 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:
    int flag=0;
    bool isBalanced(TreeNode* root)
    {
        dfs(root);
        return flag==0;
    }
    int dfs(TreeNode *root)
    {
        if(!root||flag)
            return 0;
        int L=dfs(root->left);
        int R=dfs(root->right);
        if(abs(L-R)>1)
            flag=1;
        return max(L,R)+1;
    }
};