Leetcode Balanced Binary Tree 平衡二叉树

题目：

Given a binary tree, determine if it is height-balanced.

For this problem, a height-balanced binary tree is defined as a binary tree in which the depth of the two subtrees of every node never differ by more than 1.

分析：

从二叉树的底部的节点开始检查是否平衡。二叉树的问题大部分可以用分治的思想，即一个节点如果平衡需要它的左右节点都平衡，并且它自己也需要满足条件。但是leetcode的测试用例中有一个很长的单链数据会超时。所以加一个控制，如果某一个节点的某个孩子是null，那么找另一个孩子的高度时不需要递归很深，只要看高度能不能达到2就可以。

Java代码实现：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public boolean isBalanced(TreeNode root) {
        if(root==null)
            return true;
        
        boolean flag = false;
        
        if(root.left==null || root.right==null)
            flag = true;
            
        return isBalanced(root.left) && isBalanced(root.right) && Math.abs((height(root.left, flag)-height(root.right, flag)))<=1;
    }
    
    private int height(TreeNode root, boolean flag)
    {
        if(root==null)
            return 0;
            
        if(flag)
        {
            if(root.left!=null || root.right!=null)
                return 2;
        }

        return Math.max(height(root.left, flag), height(root.right, flag))+1;
    }
}