110.バランスツリー-前順遍歴-簡単

3367 ワード

問題の説明
二叉木を与えて、高さバランスのとれた二叉木かどうかを判断します.
この問題では、高さバランスのツリーを次のように定義します.
ツリーの各ノード の左右のサブツリーの高さ差の絶対値は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 3/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;
    }
}