剑指offer-平衡二叉树

题目

题目描述
输入一棵二叉树，判断该二叉树是否是平衡二叉树。

思路

树的题目一般都是递归做法。平衡二叉树是指左右子树均为平衡二叉树，且左右字数长度之差不超过1.

代码

public class Solution {
    public boolean IsBalanced_Solution(TreeNode root) {
        return getDepth(root) != -1;
    }
     
    private int getDepth(TreeNode root) {
        if (root == null) return 0;
        int left = getDepth(root.left);
        if (left == -1) return -1;
        int right = getDepth(root.right);
        if (right == -1) return -1;
        return Math.abs(left - right) > 1 ? -1 : 1 + Math.max(left, right);
    }
}