Geek

博客园 首页 新随笔 联系 订阅 管理
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public boolean isBalanced(TreeNode root) {
        if(root == null) return true;
        height(root);
        return this.flag;

    }
    private boolean flag = true;
    // public int height(TreeNode h) {
    //     if(h==null) return 0;
    //     return 1+ Math.max(height(h.left), height(h.right));
    // }
    // public void dfs(TreeNode h) {
    //     if(h==null) return;
    //     if(flag==false) return;
    //     int left = height(h.left);
    //     int right = height(h.right);
        
    //     if(Math.abs(left-right)>1) {
    //         flag = false;
    //         return;
    //     }
    //     dfs(h.left);
    //     dfs(h.right);
    // }

    public int height(TreeNode h) {
        if(h == null || flag == false) return 0;
        int l = height(h.left)+1;
        int r = height(h.right)+1;
        if(Math.abs(l-r)>1) {
            flag = false;
        }
        return Math.max(l,r);
    }











}

 

posted on 2020-07-14 21:10  .geek  阅读(142)  评论(0编辑  收藏  举报