《剑指offer》50、二叉树操作之平衡二叉树

IceelfLuo

于 2020-12-31 09:54:54 发布

阅读量125

点赞数

分类专栏： python 文章标签：二叉树数据结构剑指offer python

本文链接：https://blog.csdn.net/iceelfluo/article/details/112003092

版权

python 专栏收录该内容

41 篇文章 1 订阅

订阅专栏

offer50的要求是：输入一棵二叉树，判断该二叉树是否是平衡二叉树。若左右子树深度差不超过1则为一颗平衡二叉树。

说白了，就是求二叉树的深度问题，深度问题就是递归问题，二叉树的递归想必我们已经写烂了。

# offer50-solution
class Solution:
    def TreeDepth(self, pRoot):  # 1、二叉树深度
        if pRoot is None:
            return 0
        return max(self.TreeDepth(pRoot.left),self.TreeDepth(pRoot.right))+1
        # 二叉树深度等于左右子树较深者的深度+1，以此递归

    def IsBalanced_Solution(self, pRoot):
        if not pRoot:
            return True
        ldepth = self.TreeDepth(pRoot.left)
        rdepth = self.TreeDepth(pRoot.right)
        if abs(ldepth-rdepth) <= 1:
            return True
        else:
            return False