leetcode110.平衡二叉树

98 阅读1分钟

110.平衡二叉树

给定一个二叉树,判断它是否是高度平衡的二叉树。本题中,一棵高度平衡二叉树定义为:

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。

给定二叉树 [1,2,2,3,3,null,null,4,4]

       1
      / \
     2   2
    / \
   3   3
  / \
 4   4               返回 false
class Solution:
    def isBalanced(self, root: TreeNode) -> bool:
        def depth(p):
            if not p:
                return 0
            left = depth(p.left)
            right = depth(p.right)
            if left == -1 or right == -1 or abs(left - right) > 1:
                return -1
            return 1 + max(left, right)

        if depth(root) == -1:
            return False
        return True