题目介绍
给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。
示例 1:
输入:root = [3,9,20,null,null,15,7] 输出:true 示例 2:
输入:root = [1,2,2,3,3,null,null,4,4] 输出:false 示例 3:
输入:root = [] 输出:true
提示:
树中的节点数在范围 [0, 5000] 内 -104 <= Node.val <= 104
来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/ba… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
定义一个递归方法来获取节点高度,在获取高度的同时比较左右子树的高度差,如果是大于1的返回-2,所以提前再判断如果左右子树有一个小于0的也返回-2,递归方法如果最终结果是大于等于0,说明层级差不超过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 isBalanced = function(root) {
function getHeight(root) {
if (!root) return 0
let l = getHeight(root.left)
let r = getHeight(root.right)
if (l < 0 || r < 0) return -2
if (Math.abs(l - r) > 1) return -2
return Math.max(l, r) + 1
}
return getHeight(root) >= 0
};