题目描述

输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。

测试用例

给定二叉树 [3,9,20,null,null,15,7]

    3
   / \
  9  20
    /  \
   15   7

返回 true

代码实现

后续遍历记录高度, 只要有不平衡了，就返回-1，不用再分析了

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @return {boolean}
 */
var isBalanced = function(root) {
    var getDepth = (root) => {
        if(!root){
            return 0;
        }

        let left = getDepth(root.left);
        if(left === -1){
            return -1;
        }
        let right = getDepth(root.right);
        if(right === -1){
            return -1;
        }

        if(Math.abs(left - right) > 1){
            return -1;
        }
        else{
            return 1 + Math.max(left, right);
        }
    }
    return getDepth(root) === -1 ? false : true;
};