剑指offer—平衡二叉树

题目：

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

题目要求判断一颗二叉树是否是平衡树，按照课本定义，平衡二叉树为，任意两个子树的深度差不超过1。
因此本质上是求取二叉树的高度，比较各个子树之间的高度差。

1. 递归版本

引用之前的求解二叉树深度的题目：二叉树深度，只需要理解了深度题目，就可以很好的解决本题。

public boolean IsBalanced_Solution(TreeNode root) {
        if(root == null){
          return true;
        }
        int left = depth(root.left);
        int right = depth(root.right);
        int distance =  right - left ;
        // 如果不是，则快速返回
        if(distance > 1 || distance < -1){
            return false;
        }
        return IsBalanced_Solution(root.left) && IsBalanced_Solution(root.right);
    }
    
    // 求解每颗子树的深度
    public int depth(TreeNode root){
        if(root == null){
            return 0;
        }
        return Math.max(depth(root.left),depth(root.right))+1;
    }
}

上面是递归方法，尤其注意下面这个if判断，如果成立，则不需要继续判断了，直接返回false，但是该方法会重复遍历子树，因为整个遍历过程是自顶向底遍历的，判断子树是否是平衡二叉树。

那么是否可以做到自底向顶遍历呢？只要底部子树不是平衡二叉树，则直接返回false，如果是平衡二叉树，则往上层遍历。

public class Solution {
    public boolean IsBalanced_Solution(TreeNode root) {
        return getDepth(root) != -1;
    }
    
    /**
     * 返回的值，即可以直接表示是否是平衡二叉树，又可以表示树的深度
     **/
    public int getDepth(TreeNode root){
        // 利用0 表示平衡二叉树、1表示非平衡二叉树
        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);
    }
}