JavaScript——leetcode剑指 Offer 55 - II. 平衡二叉树【搜索与回溯算法专题】

题目描述

输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过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 。

代码片

/*
 * @lc app=leetcode.cn id=110 lang=javascript
 *
 * [110] 平衡二叉树
 */

// @lc code=start
/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {boolean}
 */

/**
 * 
 * 只计算一次最大深度，计算的过程中在后序遍历位置顺便判断二叉树是否平衡
 * 对于每个节点，先算出来左右子树的最大高度，
 * 然后在后续遍历的位置根据左右子树的最大高度判断平衡性。
 */

var isBalanced = function(root) {
    //记录二叉树是否平衡
    let flag = true;
    const maxDepth = (root) => {
        if(root == null) return 0;
        let leftMaxDepth = maxDepth(root.left);
        let rightMaxDepth = maxDepth(root.right);
        //如果左右最大深度大于1，就不是平衡二叉树
        if(Math.abs(rightMaxDepth - leftMaxDepth) > 1) {
            flag = false;
        }
        return 1 + Math.max(leftMaxDepth, rightMaxDepth);
    };
    maxDepth(root);
    return flag;
};
// @lc code=end