Balanced Binary Tree

https://leetcode.com/problems/balanced-binary-tree/

 

以前应该已经做过这类题了

 

已经蠢到不忍直视。。

 

1. 遍历

对于左右孩子分别判断是否是平衡的，并且要求左右高度差为<2， 显然在求高度的时候，有些许许多多的重复操作

2. 也是遍历，但是避免重复操作

 

在求高度的时候，可以将是否是平衡树的信息以数字形式return回去，如果任意左，右孩子的值为-1则说明这个分支已经不平衡了，那么当前节点也return -1就可以，否则，再判断是否平衡；如果平衡的话，再return 1 + Max(left, right) 就可以了。

/**
 * 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;
        int val = getHeight2(root);
        if(val == -1) {
            return false;
        } else {
            return true;
        }
        // int left = getHeight(root.left);
        // int right = getHeight(root.right);
        // return isBalanced(root.left) && isBalanced(root.right) && Math.abs(left - right) < 2;
    }
    
    private int getHeight(TreeNode node) {
        if(node == null) return 0;
        int left=0, right=0;
        if(node.left != null) {
            left =  getHeight(node.left);
        }
        if(node.right != null) {
            right =  getHeight(node.right);
        }
        
        return 1 + Integer.max(left, right);
    }
    
    private int getHeight2(TreeNode node) {
        if(node == null) return 0;
        int left = getHeight2(node.left);
        int right= getHeight2(node.right);
        if( left == -1 || right == -1 || Math.abs(left - right) > 1) {
            return -1;
        } else {
            return 1 + Integer.max(left, right);
        }
    }
}