[leetcode] 110. 平衡二叉树

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

示例 1：

输入：root = [3,9,20,null,null,15,7]
输出：true

示例 2：

输入：root = [1,2,2,3,3,null,null,4,4]
输出：false

示例 3：

输入：root = []
输出：true

提示：

• 树中的节点数在范围 [0, 5000] 内
• -104 <= Node.val <= 104

Python实现

这道题虽然是leetcode的easy的题目，但是写出来还是不容易，平衡二叉树的每一层都需要判断左右子树的高度差，如果知道这个就需要写一个求高度的函数，还有一个递归遍历左右子树的函数，写完这些就基本上差不多了。

# 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 getDepth(self, root):
        if not root:
            return 0
        return max(self.getDepth(root.left), self.getDepth(root.right))+1

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