一、题目内容 简单

给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为: :::info 一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。 :::

示例1:

输入:root = [3,9,20,null,null,15,7] 输出:true 7. 平衡二叉树(110) - 图1

示例2:

输入:root = [1,2,2,3,3,null,null,4,4] 输出:false 7. 平衡二叉树(110) - 图2

示例 3:

输入:root = [] 输出:true

提示:

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

    二、解题思路

    这里强调一波概念:

  • 二叉树节点的深度:指从根节点到该节点的最长简单路径边的条数。

  • 二叉树节点的高度:指从该节点到叶子节点的最长简单路径边的条数。

但 leetcode 中强调的深度和高度很明显是按照节点来计算的,如图:
image.png
我们用递归计算的思路。递归嘛,化归到最少的元素,就 [4, 3 ,5]
image.png
首先,从 4 开始,递归到 3 。那 3 往下了,没有 left、right,那么可以知道 3 的高度是 0。
返回,回到 4,拿到了左子树的高度,现在拿右子树的高度,同理,知道 5 的高度是 0。
回到 4,开始比较左右子树的高度差,是否小于等于 1。如果符合,那么该树是平衡二叉树。

  1. const dfs = (node) => {
  2. if (!node) return 0
  3. const leftHeight = dfs(node.left)
  4. const rightHeight = dfs(node.right)
  5. if (Math.abs(leftHeight - rightHeight) < 2) return Math.max(leftHeight, rightHeight) + 1
  6. // 不符合平衡二叉树,就会走到这里
  7. }

但是等等,这样就完了吗,如果,下面拿到了不符合的情况,怎么向上层传递?
如果不符合,我们需要给点标志吧,那就用 -1 吧。 当接受到 -1,就让他一直往上冒泡就好了。

  1. const dfs = (node) => {
  2. if (!node) return 0
  3. const leftHeight = dfs(node.left)
  4. if (leftHeight === -1) return -1 // 往上冒泡
  5. const rightHeight = dfs(node.right)
  6. if (rightHeight === -1) return -1 // 往上冒泡
  7. if (Math.abs(leftHeight - rightHeight) < 2) return Math.max(leftHeight, rightHeight) + 1
  8. // 不符合平衡二叉树,就会走到这里
  9. return -1
  10. }

三、具体代码

  1. /**
  2. * @param {TreeNode} root
  3. * @return {boolean}
  4. */
  5. var isBalanced = function (root) {
  6. if (!root) return true
  7. const dfs = (node) => {
  8. if (!node) return 0
  9. const leftHeight = dfs(node.left)
  10. if (leftHeight === -1) return -1
  11. const rightHeight = dfs(node.right)
  12. if (rightHeight === -1) return -1
  13. if (Math.abs(leftHeight - rightHeight) < 2) return Math.max(leftHeight, rightHeight) + 1
  14. return -1
  15. }
  16. return dfs(root) === -1 ? false : true
  17. };

四、其他解法