Java版剑指offer：平衡二叉树

Java版剑指offer：平衡二叉树

描述
输入一棵二叉树，判断该二叉树是否是平衡二叉树。
在这里，我们只需要考虑其平衡性，不需要考虑其是不是排序二叉树
平衡二叉树（Balanced Binary Tree），具有以下性质：它是一棵空树或它的左右两个子树的高度差的绝对值不超过1，并且左右两个子树都是一棵平衡二叉树。

注：我们约定空树是平衡二叉树。

• 判断一个数是否为平衡二叉树。平衡二叉树是左子树的高度与右子树的高度差的绝对值小于等于1，同样左子树是平衡二叉树，右子树为平衡二叉树。
• 根据定义，如果我们能够求出以每个结点为根的树的高度，然后再根据左右子树高度差绝对值小于等于1，,就可以判断以每个结点为根的树是否满足定义。

public class Solution {
    public boolean IsBalanced_Solution(TreeNode root) {
        //正常算树的深度，中间如果有左右子树高度差大于1就返回-1
        //遇到-1必须提前返回，不能传递给上面的树，否则一定出错
        return TreeDepth(root) != -1;
        
    }
    
    public int TreeDepth(TreeNode root){
        if(root == null)
            return 0;
        //如果左右子树有不平衡的，直接返回-1
        int left=TreeDepth(root.left);
        //如果左子树或右子树不是二叉平衡树的话，必须提前停止或剪枝，
        //否则最后return的时候计算上-1结果就错了
        if (left==-1) return -1;
        int right=TreeDepth(root.right);
        if (right==-1) return -1;
        return Math.abs(left-right)<2 ? (Math.max(left,right)+1) : -1;       
    }
}