字节-LeetCode【110. 平衡二叉树】

//给定一个二叉树，判断它是否是高度平衡的二叉树。 
//
// 本题中，一棵高度平衡二叉树定义为： 
//
// 
// 一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。 
// 
//
// 
//
// 示例 1： 
//
// 
//输入：root = [3,9,20,null,null,15,7]
//输出：true
// 
//
// 示例 2： 
//
// 
//输入：root = [1,2,2,3,3,null,null,4,4]
//输出：false
// 
//
// 示例 3： 
//
// 
//输入：root = []
//输出：true
// 
//
// 
//
// 提示： 
//
// 
// 树中的节点数在范围 [0, 5000] 内 
// -104 <= Node.val <= 104 
// 
// Related Topics 树 深度优先搜索 递归 
// 👍 627 👎 0

class Solution {
    boolean isBalanced = true;
    public boolean isBalanced(TreeNode root) {
        if (root == null) {
            return true;
        }
        dfsTree(root);
        return isBalanced;
    }

    /**
     * 思路：自下往上
     * 先遍历最左节点，计算其深度；
     * 再遍历右节点，计算其深度；
     * 最后递归返回每个节点的深度
     *
     * @param root
     * @return
     */
    private int dfsTree(TreeNode root) {

        if (root == null) {
            return 0;
        }
        int left = dfsTree(root.left) + 1;
        int right = dfsTree(root.right) + 1;
        if (left - right > 1 || right - left > 1) {
            isBalanced = false;
        }

        // 当前节点的深度是最大的左右子树深度
        return Math.max(left, right);
    }
}