Leetcode 110 平衡二叉树（从底至顶，与从顶至底，递归思想及合理剪枝）

Leetcode 110 平衡二叉树（从底至顶，与从顶至底）

题目难度：简单

又一次被二叉树问题绕进入了，对于二叉树的高度和深度的编写一直是问题。因此以本题为例分析一下二叉树的深度与高度问题。

首先我们需要明确什么是高度什么是深度，我们以算法第四版为例：

• 某节点的深度：从根节点到该节点的边数上
• 某节点的高度：从该节点到叶节点的最长边数

注意：叶节点是至没有任何子节点的节点，高度为0

• 高度的计算：

1. 当root节点为空时，返回节点高度为0；
2. 当root节点不空时，返回左右子节点的高度的最大值+1。

一句话max(depth(root.right), depth(root.left))+1。

• 深度的计算

1. 当 root 节点为空时，返回 0
2. 当 root 节点左右孩子有一个为空时，返回不为空的孩子节点的深度+1
3. 当 root 节点左右孩子都不为空或都空时，返回左右孩子较小深度的节点值+1 （如果都空，参照1都为0）

一句话min(height(root.left), height(root.right))+1。

从底至顶（提前阻断，比起从顶到下合理剪枝）

此方法为本题的最优解法，但“从底至顶”的思路不易第一时间想到。

思路是对二叉树做先序遍历，从底至顶返回子树最大高度，若判定某子树不是平衡树则 “剪枝” ，直接向上返回。

算法流程：

recur(root):

• 递归返回值：

1. 当节点root 左 / 右子树的高度差 <2 ：则返回以节点root为根节点的子树的最大高度，即节点 root 的左右子树中最大高度加 1( max(left, right) + 1)
2. 当节点root 左 / 右子树的高度差 ≥2 ：则返回 −1 ，代表此子树不是平衡树 。

• 递归终止条件：

1. 当越过叶子节点时，返回高度 0 ；
2. 当左（右）子树高度 left== -1 时，代表此子树的 左（右）子树 不是平衡树，因此直接返回 −1 ；

isBalanced(root) ：

• 返回值： 若 recur(root) != 1 ，则说明此树平衡，返回 true ； 否则返回 false。

复杂度分析：

• 时间复杂度 O(N)： N 为树的节点数；最差情况下，需要递归遍历树的所有节点。
• 空间复杂度 O(N)： 最差情况下（树退化为链表时），系统递归需要使用 O(N) 的栈空间。

class Solution:
    def isBalanced(self, root: TreeNode) -> bool:
        return self.recur(root) != -1

    def recur(self, root):
        if not root: return 0
        left = self.recur(root.left)
        if left == -1