【剑指Offer】 39.平衡二叉树 python实现

题目描述

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

平衡二叉树：

模拟一下递归函数如何运行
不是平衡二叉树的运行过程

解题思路

递归法

python 代码

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def __init__(self):
        self.res = True
    def IsBalanced_Solution(self, pRoot):
        # write code here
        self.helper(pRoot)
        return self.res
         
    def helper(self,root):
        if not root:
            return 0
        # 获得当前节点的左子树的深度
        left = self.helper(root.left)
        # 获得当前节点的右子树的深度
        right = self.helper(root.right)
        if abs(left-right)>1:
            self.res = False
        return 1 + max(left,right)

s = Solution()
res = s.IsBalanced_Solution(pRoot)
print('.....',res)