剑指 Offer 55 - II. 平衡二叉树

题目描述:

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

示例:

示例 1:

给定二叉树 [3,9,20,null,null,15,7]
在这里插入图片描述
返回 true

示例 2:

给定二叉树 [1,2,2,3,3,null,null,4,4]
在这里插入图片描述
返回 false

思路:

思路:它是一棵空树或它的左右两个子树的高度差的绝对值不超过1,并且左右两个子树都是一棵平衡二叉树。遍历左子树和右子树的深度,然后比较二者差值

代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
var isBalanced = function(root) {
if(!root) return true
let left = dfs(root.left)
let right = dfs(root.right)
if(Math.abs(right - left) > 1) return false
//判断当前子树的 左子树 和 右子树 是否是平衡树;
return isBalanced(root.left) && isBalanced(root.right)
};
/** 获取深度 */
function dfs(root){
if(!root) return 0
return Math.max(dfs(root.left), dfs(root.right)) + 1
}