日拱一卒（2）——leetcode学习记录：平衡二叉树（二）

一、任务：平衡二叉树

给定一个二叉树，判断它是否是 平衡二叉树

二、思路：

1. 递归计算二叉树的深度

        a. 左右子节点存在一个，返回 1+递归（存在的子节点）

        b. 左右子节点都存在，返回 max(1+递归（左子节点），1+递归（右子节点））

        c. 左右子节点都不存在，返回1

        d.大前提，如果主节点为None，返回 0

2. 左右子节点深度差大于1，返回 False

3. 左右子节点深度差小于等于1，返回 递归 左子节点判断and右子节点判断

三、解答

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def depth(self,node):
        if node == None:
            return 0
        else:
            return max(1+self.depth(node.left),1+self.depth(node.right))

    def isBalanced(self, root: Optional[TreeNode]) -> bool:
        if root == None:
            return True
        if abs(self.depth(root.left)-self.depth(root.right)) > 1:
            return False
        else:
            return self.isBalanced(root.left) and self.isBalanced(root.right)

编程小白，所以结果比较差，见笑