【leetcode】面试题 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 。

题解： 这题我觉得可能会出错的地方就是，在判断根节点符合条件之后需要判断左右子树是否也符合条件！

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public boolean isBalanced(TreeNode root) {
        if(root == null || root.left == null && root.right == null){
            return true;
        }
        int leftlen = getHeight(root.left);
        int rightlen = getHeight(root.right);
        if(Math.abs(leftlen - rightlen) > 1){
            return false;
        }
        // 需要检查左右子树为根节点的时候是是否也符合条件
        return isBalanced(root.left) && isBalanced(root.right);
    }

    public int getHeight(TreeNode root){
        if(root == null){
            return 0;
        }
        return Math.max(getHeight(root.left), getHeight(root.right)) + 1;
    }
}