110. 平衡二叉树-前序遍历-简单

问题描述

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

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

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过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;
    }
}