[面试题 04.04][简单][DFS] 检查平衡性

题目描述

面试题 04.04. 检查平衡性

实现一个函数，检查二叉树是否平衡。在这个问题中，平衡树的定义如下：任意一个节点，其两棵子树的高度差不超过 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 。

解题思路

使用DFS, 从根节点开始, 递归的向左右结点延伸, 如果左右子树有任何一个是不平衡树, 则整体为非平衡树. 如果左右子树都是平衡树, 但双方的深度差距大于1, 也是非平衡树. 只有这些都满足, 才是平衡树.

因此, 递归函数要么返回不平衡(代码中使用-1代替), 要么返回子树的深度.

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def isBalanced(self, root: TreeNode) -> bool:
        def dfs(node):
            if node is None:
                return 0

            left = dfs(node.left)
            right = dfs(node.right)
            if left == -1 or right == -1:
                return -1
            if abs(left - right) > 1:
                return -1
            return max(left, right) + 1

        return True if dfs(root) != -1 else False