剑指offer 平衡二叉树 python

题目描述

输入一棵二叉树,判断该二叉树是否是平衡二叉树。

样例

如果是 返回True
如果不是 返回False

想法一:
遍历节点,计算左右子树的高度,在计算同时,如果有子树不符合平衡二叉树,则返回-1,这样不需要遍历所有节点。
流程图如下:
《剑指offer 平衡二叉树 python》
代码:

class Solution:
    def IsBalanced_Solution(self, pRoot):
        # write code here
        return self.fun(pRoot) is not -1

    def fun(self, node):
        if node is None:
            return 0
        left = self.fun(node.left)
        if left is -1:
            return -1
        right = self.fun(node.right)
        if right is -1:
            return -1
        if abs(left - right) > 1:
            return max(left, right) + 1
        else:
            return -1

最后

刷过的LeetCode或剑指offer源码放在Github上了,希望喜欢或者觉得有用的朋友点个star或者follow。
有任何问题可以在下面评论或者通过私信或联系方式找我。
联系方式
QQ:791034063
Wechat:liuyuhang791034063
CSDN:https://blog.csdn.net/Sun_White_Boy
Github:https://github.com/liuyuhang791034063

    原文作者:GF66
    原文地址: https://www.cnblogs.com/GF66/p/9785461.html
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞