LeetCode:110(Python)—— 平衡二叉树(简单)

LeetCode:110(Python)—— 平衡二叉树(简单)概述:给定一个二叉树,判断它是否是高度平衡的二叉树。一棵高度平衡二叉树定义为:一个二叉树每个节点的左右两个子树的高度差的绝对值不超过1。

平衡二叉树

概述:给定一个二叉树,判断它是否是高度平衡的二叉树。一棵高度平衡二叉树定义为:一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1

输入:root = [3,9,20,null,null,15,7]
输出:true

输入:root = [1,2,2,3,3,null,null,4,4]
输出:false

输入:root = []
输出:true

方法一:自顶向下的递归

思路:定义函数 height,用于计算二叉树中的任意一个节点 p 的高度。对于当前遍历到的节点,首先计算左右子树的高度,如果左右子树的高度差是否不超过 1 ,再分别递归地遍历左右子节点,并判断左子树和右子树是否平衡。

# 自顶向下的递归
class Solution:
    def isBalanced(self, root: Optional[TreeNode]) -> bool:
        def height(root: TreeNode) -> int:
            if not root:
                return 0
            return max(height(root.left), height(root.right)) + 1
        if not root:
            return True
        return abs(height(root.left) - height(root.right)) <= 1 \
        and self.isBalanced(root.left) and self.isBalanced(root.right)

方法二:自底向上的递归

思路:自底向上递归的做法类似于后序遍历,对于当前遍历到的节点,先递归地判断其左右子树是否平衡,再判断以当前节点为根的子树是否平衡。

# 自底向上的递归
class Solution:
    def isBalanced(self, root: Optional[TreeNode]) -> bool:
        def height(root: TreeNode) -> int:
            if not root:
                return 0
            left_height = height(root.left)
            right_height = height(root.right)
            if left_height == -1 or right_height == -1 \
            or abs(left_height - right_height) > 1:
                return -1
            else:
                return max(left_height, right_height) + 1
        return height(root) >= 0

总结

自底向上的递归就是及时止损,自顶向下的递归就是不管三七二十一先把所有的高度算一遍!

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。
如需转载请保留出处:https://bianchenghao.cn/35915.html

(0)
编程小号编程小号

相关推荐

发表回复

您的电子邮箱地址不会被公开。 必填项已用*标注