LeetCode面试题 04.04. 检查平衡性(python,递归)

1. 题目

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

示例 1:
给定二叉树 [3,9,20,null,null,15,7]

  3
   / \
  9  20
    /  \
   15   7

返回 true 。

来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/check-balance-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 题解

简单的dfs遍历，获得每个节点的深度，同时，使用变量记录子树是否平衡。

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

class Solution:
    def __init__(self):
        self.balance = True

    def isBalanced(self, root: TreeNode) -> bool:
        self.getDepth(root)
        return self.balance

    def getDepth(self, root):
        if not root:
            return 0
        left_depth = self.getDepth(root.left)
        right_depth = self.getDepth(root.right)
        if abs(left_depth-right_depth)>1:
            self.balance = False
        return max(left_depth, right_depth) + 1