LeetCode 110.平衡二叉树——Python实现（递归）

题目：给定一个二叉树，判断它是否是高度平衡的二叉树。一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。
思路：平衡二叉树的子树也是一颗平衡二叉树。所以，可以用递归的方法，从叶子节点开始，逐层向上检验子树。在每一次的
      节点计算力，需要计算的是以当前节点为root，其左右子树是否平衡，如果平衡就是true，否则就是false。那么如何判断
      左右子树是否平衡，就要计算左右子树的高度，也就是层数。同时，为了便于当前 root节点的上一级节点进行判断，需要返
      回当前子树的高度。

class Solution(object):
    def isBalanced(self, root: TreeNode) -> int:
        def dfs(root):
            # 如果当前节点是叶子节点的子节点了，那么这个节点要返回true,同时返回层数0
            if root is None:
                return True, 0

            # 如果不是空节点，则根据其子节点开始计算
            left_is_balanced, left_level = dfs(root.left)
            right_is_balanced, right_level = dfs(root.right)

            # 如果两边都平衡，且高度差不大于1,则标记本棵树是平衡的，且返回该树的高度
            if (left_is_balanced and right_is_balanced) and abs(left_level - right_level) < 2:
                return True, max(left_level, right_level) + 1

            # 其余情况下，说明该棵树都是不平衡的
            return False, 0

        # 如果是空树，则是平衡的
        if root is None:
            return True

        # 如果不是空树，则判断当前节点的左右两棵树是否平衡
        return dfs(root)[0]