【树-简单】剑指 Offer 55 - II. 平衡二叉树

最新推荐文章于 2023-04-02 14:21:47 发布

原创最新推荐文章于 2023-04-02 14:21:47 发布 · 88 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 同时被 2 个专栏收录

714 篇文章

订阅专栏

刷题

643 篇文章

订阅专栏

本文介绍了如何判断一棵二叉树是否为平衡二叉树，提供了两种Python方法。方法1通过计算各节点的深度并比较，方法2使用了深度优先搜索。示例展示了具体代码实现，并给出了测试用例。

【题目】
输入一棵二叉树的根节点，判断该树是不是平衡二叉树。如果某二叉树中任意节点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。

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

/
9 20
/
15 7

返回 true 。

示例 2:
给定二叉树 [1,2,2,3,3,null,null,4,4]

3 3
/
4 4

返回 false 。
限制：

0 <= 树的结点个数 <= 10000

注意：本题与主站 110 题相同.
【代码】
【方法1】
在这里插入图片描述

class Solution:
    def isBalanced(self, root: TreeNode) -> bool:
        def height(root):
            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)

【方法2】
在这里插入图片描述

class Solution:
    def visit(self,root):
        if not root:
            return 0
        L=self.visit(root.left)
        R=self.visit(root.right)
        if L==-1 or R==-1 or abs(L-R)>1:
            return -1
        return max(L,R)+1

    def isBalanced(self, root: TreeNode) -> bool:
        return self.visit(root)>=0