LeetCode 110. 平衡二叉树

作者:Choi Yang

更新于:7 个月前

字数统计:351 字

阅读时长:1 分钟

题目描述

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

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

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 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 。

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

解题思路

dfs，平衡二叉树就是每个节点 的左右两个子树的高度差的绝对值不超过 1。那么，我们可以自底向上，即采用后序遍历的方式，只要左右高度超过 1 了，直接设置 flase即可。

/**
 * 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) {
  if (!root) return true;
  let res = true;
  let dfs = (root) => {
    // 先遍历左子树，在遍历右子树
    let left = root.left && dfs(root.left) + 1;
    let right = root.right && dfs(root.right) + 1;
    // 判断是否是平衡二叉树，就看高度差是否大于1
    if (Math.abs(left - right) > 1) res = false;
    // 返回左右子树深度的最大值
    return Math.max(left, right);
  };
  dfs(root);
  return res;
};

学如逆水行舟，不进则退

Contributors

Choi Yang