平衡二叉树剑指offer python

最新推荐文章于 2020-07-02 09:16:21 发布

酸乳酸乳

最新推荐文章于 2020-07-02 09:16:21 发布

阅读量239

点赞数

CC 4.0 BY-SA版权

分类专栏：剑指offer

本文链接：https://blog.youkuaiyun.com/qq_41359265/article/details/84224485

剑指offer 专栏收录该内容

20 篇文章

订阅专栏

本文探讨了两种检测二叉树是否平衡的算法实现：从上向下遍历和从下向上遍历。从上向下遍历可能导致重复判断，而从下向上遍历则能有效避免这一问题。文章通过具体代码对比了两种方法，并解释了如何优化算法以提高效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

第一次写是从上向下遍历，但是这会导致很多不必要的判断，代码如下：

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def IsBalanced_Solution(self, pRoot):
        # write code here
        if not pRoot:
            return True
        if abs(self.treeDepth(pRoot.left) - self.treeDepth(pRoot.right))<=1 and self.IsBalanced_Solution(pRoot.left) and self.IsBalanced_Solution(pRoot.right):
            return True
        return False
        
        
        
    def treeDepth(self,pRoot):
        if not pRoot:
            return 0
        left = self.treeDepth(pRoot.left)
        right = self.treeDepth(pRoot.right)
        return max(left,right) + 1

如果改为从下向上遍历就可以很好的解决这个问题,，就不会出现重复判断的问题了：

class Solution:
    def IsBalanced_Solution(self, pRoot):
        # write code here
        return self.getDepth(pRoot) != -1
    
    def getDepth(self,pRoot):
        if not pRoot:
            return 0
        left = self.getDepth(pRoot.left)
        if left == -1:
            return -1
        right = self.getDepth(pRoot.right)
        if right == -1:
            return -1
        if abs(left - right) > 1:
            return -1
        return max(left,right) + 1

只是不明白为什么不用比较值之间的关系。。