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

77 阅读1分钟

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

输入一棵二叉树的根节点,判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过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 

限制:
0 <= 树的结点个数 <= 10000
注意:本题与主站 110 题相同:https://leetcode-cn.com/problems/balanced-binary-tree

解题思路:

首先获取平衡二叉树的深度,判断左右子树的深度之差是不是<=1,并且需要判断这棵二叉树是不是同时具有左右子树
Math.abs(x)=|x| 取某个数的绝对值

代码实现

var isBalanced = function(root) {
    function getHeight(root){
        if (!root) return 0
        return Math.max(getHeight(root.left), getHeight(root.right)) + 1
    }
    if (!root) return true
    return isBalanced(root.left) && isBalanced(root.right) && Math.abs(getHeight(root.left) -  getHeight(root.right)) <=1

};