题目描述
输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过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 。
代码片
/*
* @lc app=leetcode.cn id=110 lang=javascript
*
* [110] 平衡二叉树
*/
// @lc code=start
/**
* 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) {
//记录二叉树是否平衡
let flag = true;
const maxDepth = (root) => {
if(root == null) return 0;
let leftMaxDepth = maxDepth(root.left);
let rightMaxDepth = maxDepth(root.right);
//如果左右最大深度大于1,就不是平衡二叉树
if(Math.abs(rightMaxDepth - leftMaxDepth) > 1) {
flag = false;
}
return 1 + Math.max(leftMaxDepth, rightMaxDepth);
};
maxDepth(root);
return flag;
};
// @lc code=end