剑指Offer——II平衡二叉树

class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

# 这道题使用中序遍历加上剪枝的方法来做的。
class Solution:
    def isBalanced(self, root: TreeNode) -> bool:
        if self.dfs(root) == -1:return False
        else:return True
    def dfs(self,root):
        # 如果节点为空就返回0
        if not root:return 0
        # 遍历左节点
        left = self.dfs(root.left)
        # 如果左节点不是平衡二叉树，直接返回
        if left == -1:return -1
        # 遍历右节点，同理
        right = self.dfs(root.right)
        if right == -1:return -1
        # 这里就保证了左右节点都是平衡二叉树，所以就判断当前root是不是平衡二叉树。
        return max(left,right) + 1 if abs(left - right) <= 1 else -1