問題描述

給定一個二叉樹，判斷它是否是高度平衡的二叉樹。

本題中，一棵高度平衡二叉樹定義為：

一個二叉樹每個節點的左右兩個子樹的高度差的絕對值不超過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
返回false 。

來源：力扣（LeetCode）
連結：https://leetcode-cn.com/problems/balanced-binary-tree

解答

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 
 
*/
 /*
 
 前序遍歷，自底向上，時間複雜度為O(n)
 
 */
class height{
    int h;
    boolean flag;

    public height(int h, boolean flag){
        this.h = h;
        this.flag = flag;
    }
}
class Solution {
    public static height judge(TreeNode root){
        if(root == null)return new height(-1, true);
        height left 
 
= judge(root.left);
        if(!left.flag)return new height(-1,false);
        height right = judge(root.right);
        if(!right.flag)return new height(-1,false);

        if(Math.abs(left.h-right.h) > 1)return new height(-1,false);
        return new height(Math.max(left.h,right.h)+1,true);
    }
    
 
public boolean isBalanced(TreeNode root) {
        return judge(root).flag;
    }
}