Pengjielee's blog

题目

给定一个二叉树，判断它是否是高度平衡的二叉树。

本题中，一棵高度平衡二叉树定义为：

一个二叉树每个节点的左右两个子树的高度差的绝对值不超过 1 。

示例 1：
            3
          /   \
         9    20
             /  \
           15    7

输入：root = [3,9,20,null,null,15,7]
输出：true

示例 2：
            1
           / \
          2   2
         / \
        3   3
       / \
      4   4   
输入：root = [1,2,2,3,3,null,null,4,4]
输出：false

示例 3：
输入：root = []
输出：true

提示：

树中的节点数在范围 [0, 5000] 内 -10^4 <= Node.val <= 10^4

来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/balanced-binary-tree 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

JS实现

参考1：自顶向下的递归

/**
 * 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 height = (root) => {
  if (root === null) {
    return 0;
  } else {
    return Math.max(height(root.left), height(root.right)) + 1;
  }
};
var isBalanced = function (root) {
  if (root === null) {
    return true;
  } else {
    return (
      Math.abs(height(root.left) - height(root.right)) <= 1 &&
      isBalanced(root.left) &&
      isBalanced(root.right)
    );
  }
};

参考2：自底向上的递归

var height = (root) => {
  if (root == null) {
    return 0;
  }
  const leftHeight = height(root.left);
  const rightHeight = height(root.right);
  if (
    leftHeight === -1 ||
    rightHeight === -1 ||
    Math.abs(leftHeight - rightHeight) > 1
  ) {
    return -1;
  } else {
    return Math.max(leftHeight, rightHeight) + 1;
  }
};
var isBalanced = function (root) {
  return height(root) >= 0;
};