平衡二叉树的定义:左右子树高度差的绝对值 <= 1,且左右子树都是平衡二叉树

显然,具有递归特性,考虑使用递归。

此时,需要计算左右子树的高度差,以及左右子树各自是否是平衡二叉树,显然一个是int,一个是boolean

需要确定递归过程或者说递归返回值,是递归深度int,还是递归是否是平衡二叉树boolean,由于是否是平衡二叉树是结果,判断过程依赖于深度,且递归boolean不便于确定递归终止条件,所以主要递归深度。

考虑将深度int转化为是否为平衡二叉树的boolean,可以 设非平衡二叉树的深度返回值为-1,平衡二叉树返回非-1的具体深度值

public class Solution {
    public boolean IsBalanced_Solution(TreeNode root) {

        if (root == null) {
            return true;
        }
        return depthDiff(root) == -1 ? false : true;
    }

    public int depthDiff(TreeNode root) {
        if (root == null) {
            return 0;
        }

        int left = depthDiff(root.left);
        int right = depthDiff(root.right);
        // 如果有一方等于-1,直接返回-1,因为左右子树已经有一方不是平衡二叉树了
        if (left == -1 || right == -1) {
            return -1;
        }
        // > 1,说明已经不是平衡二叉树了
        if (Math.abs(left - right) > 1) {
            return -1;
        } else {
            // 如果是平衡二叉树,返回较大深度
            return Math.max(left, right) + 1;
        }
    }
}