题解 | #判断是不是平衡二叉树#

判断是不是平衡二叉树

https://www.nowcoder.com/practice/8b3b95850edb4115918ecebdf1b4d222

题目:https://www.nowcoder.com/practice/8b3b95850edb4115918ecebdf1b4d222
递归遍历二叉树的每一个节点,求出节点左右子树的高度,如果高度之差不超过1,则为平衡二叉树。先求高度(递归求),再判断是不是平衡二叉树(递归判断)。

# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
#
# 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
#
# 
# @param pRoot TreeNode类 
# @return bool布尔型
#
class Solution:
    def TreeDepth(self , pRoot: TreeNode) -> int:
        if pRoot==None:
            return 0
        else:
            return max(self.TreeDepth(pRoot.left),self.TreeDepth(pRoot.right))+1 

    def IsBalanced_Solution(self , pRoot: TreeNode) -> bool:
        # 递归遍历二叉树的每一个节点,求出节点左右子树的高度,如果高度之差不超过1,则为平衡二叉树
        if not pRoot:
            return True #空树是平衡二叉树
       #递归求左右子树高度,所以要写递归求高度的函数
        ldepth=self.TreeDepth(pRoot.left)
        rdepth=self.TreeDepth(pRoot.right)
        diff=ldepth-rdepth
        if abs(diff)>1:
            return False
        return self.IsBalanced_Solution(pRoot.left) and self.IsBalanced_Solution(pRoot.right)
全部评论

相关推荐

11-30 11:07
河南大学 Java
宇宙厂 测开 n*15
丘丘给个offer:有后选后
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务