平衡二叉树

110. 平衡二叉树

难度简单291

给定一个二叉树，判断它是否是高度平衡的二叉树。

本题中，一棵高度平衡二叉树定义为：

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。

示例 1:

给定二叉树 [3,9,20,null,null,15,7]

    3
   / \
  9  20
    /  \
   15   7

返回 true 。

/**

//注意：这里需要返回左子树是否平衡，右子树是否平衡，和左右子树的高度的最大值，布尔类型的判断res需要使用布尔类型的数组，只有数组才会在调用函数的时候改变值。

 * 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) {

        boolean[] res = new boolean[1];

        res[0] = true;

        getHeight(root,1,res);

        return res[0];

    }

    public int getHeight(TreeNode root,int level,boolean[] res){

        if(root == null){

            return level;

        }

        int lH = getHeight(root.left, level+1, res);

        if(!res[0]){

            return level;

        }

        int rH = getHeight(root.right, level+1, res);

        if(!res[0]){

            return level;

        }

        if(Math.abs(rH - lH) > 1){

            res[0] = false;

        }

        return Math.max(rH, lH);

    }

}