[剑指offer] 55-2. 平衡二叉树

题目

输入一棵二叉树的根结点，判断该树是不是平衡二叉树。
如果某二叉树中任意结点的左右子树的深度相差不超过1，那么它就是一棵平衡二叉树。
注意：规定空树也是一棵平衡二叉树。

v1：后序遍历+剪枝

后序遍历，从底至顶，返回子树深度，若判定某子树不是平衡树则 “剪枝” ，即，直接返回。

depth(root) 函数：计算树root的深度+判断子树平衡
返回值：
当节点root 左 / 右子树的深度差≤1 ：则返回当前子树的深度，即节点 root 的左 / 右子树的深度最大值 +1；
当节点root 左 / 右子树的深度差>2 ：则返回-1 ，代表此子树不是平衡树 。
终止条件：
当 root 为空：说明越过叶节点，因此返回高度 0；
当左（右）子树深度为 −1 ：代表此树的 左（右）子树 不是平衡树，因此剪枝，直接返回 -1 ；(也可以写进返回值条件里，是一样的)

isBalanced(root) 主函数：
返回值： 若 recur(root) != -1 ，则说明此树平衡，返回 True ； 否则返回 False 。

class Solution(object):
    def isBalanced(self, root):
        if not root:return True
        def depth(root):
            if not root:
                return 0
            left = depth(root.left)
            # if left == -1: return -1
            right = depth(root.right)
            # if right == -1: return -1
            return max(left,right)+1 if abs(left-right)<2 and left!=-1 and right!=-1 else -1
        return depth(root)!=-1

v2：先序遍历+判断深度

先序遍历，自顶向底，大量冗余计算。

depth(root) 函数： 计算树 root 的深度
终止条件： 当 root 为空，即越过叶子节点，则返回高度 0 ；
返回值： 返回左 / 右子树的深度的最大值 +1。

isBalanced(root) 主函数： 判断树 root 是否平衡
特例处理： 若树根节点 root 为空，则直接返回 True ；
返回值： 所有子树都需要满足平衡树性质，因此以下三者使用与逻辑 and 连接；
abs(self.depth(root.left) - self.depth(root.right)) <= 1 ：判断 当前子树 是否是平衡树；
self.isBalanced(root.left) ： 先序遍历递归，判断 当前子树的左子树 是否是平衡树；
self.isBalanced(root.right) ： 先序遍历递归，判断 当前子树的右子树 是否是平衡树；

class Solution:
    def isBalanced(self, root: TreeNode) -> bool:
        if not root: return True
        return abs(self.depth(root.left) - self.depth(root.right)) < 2 and \
            self.isBalanced(root.left) and self.isBalanced(root.right)

    def depth(self, root):
        if not root: return 0
        return max(self.depth(root.left), self.depth(root.right)) + 1