Day 17 二叉树part04： 110.平衡二叉树 257. 二叉树的所有路径 404.左叶子之和

110. 平衡二叉树

后序遍历

给定一个二叉树，判断它是否是高度平衡的二叉树。

本题中，一棵高度平衡二叉树定义为：

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。

示例 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

平衡二叉树定义为：一个二叉树每个节点的左右两个子树的高度差的绝对值不超过 1。

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    def isBalanced(self, root):
        """
        :type root: TreeNode
        :rtype: bool
        time : O(n)
        space: 最好 O(log n)  最坏 O(n)
        """
        
        if self.get_height(root) == -1:
            return False
        else:
            return True

       
    def get_height(self, root):
        # Base Case
        if not root:
            return 0

        # 左
        left_height = self.get_height(root.left)
        if left_height == -1:
            return -1

        # 右
        right_height = self.get_height(root.right)
        if right_height == -1:
            return -1

        # 中
        if abs(left_height - right_height) > 1:
            return -1
        else:
            return 1 + max(left_height, right_height)
        

代码分析：

1. isBalanced 函数是主要的接口，它会返回 True 或 False，表示树是否平衡。

   • 这个函数通过调用辅助函数 get_height 来获取树的高度。如果 get_height 返回 -1，表示该树不平衡，因此返回 False；否则返回 True。

2. get_height 函数是一个递归函数，用于计算二叉树的高度。

   • 如果当前节点不存在（即为 None），则返回高度 0。
   • 对当前节点的左子树和右子树分别递归调用 get_height。
     • 如果左子树或右子树返回 -1，则立即返回 -1，表示该子树不平衡。
   • 如果左子树和右子树的高度差的绝对值大于 1，则返回 -1，表示整个树不平衡。
   • 否则，返回当前树的高度，即左子树和右子树的最大高度加 1。

总之，这段代码的核心思想是：在计算树的高度的过程中，如果发现某个子树的左右子树高度差大于 1，立即返回 -1，这样可以提早结束不必要的计算，并且快速确定树是否平衡。

关于 := 符号：

• 在 Python 3.8 中，引入了一个新的语法，称为“海象操作符”或“赋值表达式”（Assignment Expressions），它允许你在表达式中进行赋值。这个符号可以使某些代码更加简洁，避免重复的计算或调用。

例如，在这段代码中：

if (left_height := self.get_height(root.left)) == -1:

left_height := self.get_height(root.left) 将 self.get_height(root.left) 的返回值赋给 left_height，同时这个表达式的值就是 self.get_height(root.left) 的返回值，因此可以直接与 -1 进行比较。

使用 :=（赋值表达式，海象操作符）主要是为了避免多次调用相同的函数或重复的计算，从而使代码更简洁。但这并不意味着你必须使用它。如果你认为普通的方式（如你所示的代码）更容易阅读和理解，完全可以这样写。

以下是使用普通的 if 语句来写的代码段：

left_height = self.get_height(root.left)
if left_height == -1:
    return -1

257. 二叉树的所有路径

给你一个二叉树的根节点 root ，按 任意顺序 ，返回所有从根节点到叶子节点的路径。

叶子节点 是指没有子节点的节点。

 

示例 1：

输入：root = [1,2,3,null,5]
输出：["1->2->5","1->3"]

示例 2：

输入：root = [1]
输出：["1"]

提示：

• 树中节点的数目在范围 [1, 100] 内
• -100 <= Node.val <= 100

 

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution(object):
    #  迭代法 
    def binaryTreePaths(self, root):
        """
        :type root: TreeNode
        :rtype: List[str]
        time: O(n)
        space: O(n)
        """
        if not root:
            return []
        
        # 栈用于深度优先遍历
        stack = [(root, str(root.val))]
        res = [] # 返回最终路径

        while stack:
            node, path = stack.pop()

            # 如果是叶子节点，则保存路径
            if not node.left and not node.right:
                res.append(path)

            # 如果左子节点存在，更新路径并入栈
            if node.left:
                stack.append((node.left, path + '->' + str(node.left.val)))
            
            # 如果右子节点存在，更新路径并入栈
            if node.right:
                stack.append((node.right, path + '->' + str(node.right.val)))
            
        return res