【树-简单】面试题 04.04. 检查平衡性

最新推荐文章于 2024-02-10 07:45:00 发布

原创最新推荐文章于 2024-02-10 07:45:00 发布 · 109 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 同时被 2 个专栏收录

714 篇文章

订阅专栏

刷题

643 篇文章

订阅专栏

本文探讨了如何实现并优化二叉树平衡性的检查，通过实例和三种不同的方法展示了如何递归地计算节点高度差，并判断是否符合平衡定义。理解这些算法有助于提高对二叉树结构的理解和代码编写能力。

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

示例 1:

给定二叉树 [3,9,20,null,null,15,7]
3
/
9 20
/
15 7
返回 true 。

示例 2:

给定二叉树 [1,2,2,3,3,null,null,4,4]
1
/
2 2
/
3 3
/
4 4
返回 false 。
【代码】
【方法1】
在这里插入图片描述

class Solution:
    def visit(self,root):
        if not root:
            return 0
        L=self.visit(root.left)
        R=self.visit(root.right)
        if abs(L-R)>1:
            self.flag=False
        return max(L,R)+1
    def isBalanced(self, root: TreeNode) -> bool:
        self.flag=True
        self.visit(root)
        return self.flag

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

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

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

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)