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]

 

你可能感兴趣的:(数据结构与算法,数据结构与算法,LeetCode,二叉树,平衡二叉树)