【算法】平衡二叉树

题目描述

给定一个二叉树，判断它是否是高度平衡的二叉树。

本题中，一棵高度平衡二叉树定义为：

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。

示例 1:

给定二叉树 [3,9,20,null,null,15,7]

    3
   / \
  9  20
    /  \
   15   7

返回 true 。

示例 2:

给定二叉树 [1,2,2,3,3,null,null,4,4]

       1
      / \
     2   2
    / \
   3   3
  / \
 4   4

解题

思路：要想知道一棵树是否是平衡的，需要满足左右子树高的差小于等于1，且左右子树分别为平衡二叉树。

方法一：自顶向下
其实就是正向思考过程，先获取左子树高，再获取右子树高，比较大小；再递归判断左、右子树是否平衡，执行步骤不变。

class Solution {
    public boolean isBalanced(TreeNode root) {
        if(root==null) return true;
        int m = height(root.left);
        int n = height(root.right);
        return Math.abs(m-n)<=1 && isBalanced(root.left) && isBalanced(root.right);
    }
    public int height(TreeNode r){
        if(r==null) return 0;
        int m = height(r.left);
        int n = height(r.right);
        return Math.max(m, n)+1;
    }
}

方法二：自底向上
方法一的方法有大量冗余，获取树高时遍历子树一遍，判断子树平衡时又遍历一遍。
所以考虑进行一次遍历就能获取树高，和树是否平衡。
从叶子节点开始，先判断树是否平衡，若平衡向上层返回树高，若不平衡则直接返回不平衡结果，中止调用。

class Solution {
    public boolean isBalanced(TreeNode root) {
        return helper(root)!=-1;
    }
    public int helper(TreeNode root){
        if(root==null) return 0;
        int l = helper(root.left);
        if(l==-1) return -1;
        int r = helper(root.right);
        if(r==-1) return -1;
        if(Math.abs(l-r)<=1) 
            return Math.max(l, r)+1;
        return -1;
    }
}