leetcode[872]叶子相似的树 Python3实现 (dfs获得二叉树叶子的集合）

# 请考虑一棵二叉树上所有的叶子，这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。 
# 
#  
# 
#  举个例子，如上图所示，给定一棵叶值序列为 (6, 7, 4, 9, 8) 的树。 
# 
#  如果有两棵二叉树的叶值序列是相同，那么我们就认为它们是 叶相似 的。 
# 
#  如果给定的两个根结点分别为 root1 和 root2 的树是叶相似的，则返回 true；否则返回 false 。 
# 
#  
# 
#  示例 1： 
# 
#  
# 
#  
# 输入：root1 = [3,5,1,6,2,9,8,null,null,7,4], root2 = [3,5,1,6,7,4,2,null,null,nul
# l,null,null,null,9,8]
# 输出：true
#  
# 
#  示例 2： 
# 
#  
# 输入：root1 = [1], root2 = [1]
# 输出：true
#  
# 
#  示例 3： 
# 
#  
# 输入：root1 = [1], root2 = [2]
# 输出：false
#  
# 
#  示例 4： 
# 
#  
# 输入：root1 = [1,2], root2 = [2,2]
# 输出：true
#  
# 
#  示例 5： 
# 
#  
# 
#  
# 输入：root1 = [1,2,3], root2 = [1,3,2]
# 输出：false
#  
# 
#  
# 
#  提示： 
# 
#  
#  给定的两棵树可能会有 1 到 200 个结点。 
#  给定的两棵树上的值介于 0 到 200 之间。 
#  
#  Related Topics 树 深度优先搜索 
#  👍 104 👎 0

# leetcode submit region begin(Prohibit modification and deletion)
# 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 leafSimilar(self, root1: TreeNode, root2: TreeNode) -> bool:
        def dfs(root: TreeNode, l):
            if root.left:
                dfs(root.left, l)
            if root.right:
                dfs(root.right, l)
            if not root.left and not root.right:
                l.append(root.val)

        l1 = []
        l2 = []
        dfs(root1, l1)
        dfs(root2, l2)
        return True if l1 == l2 else False
# leetcode submit region end(Prohibit modification and deletion)

需按从左到右的顺序拿到二叉树的叶子节点，用dfs；

若是需按层次顺序，则可用bfs；

为了保存生成的叶子节点序列，将空数组l作为参数传递给dfs函数，结果都存在l里。官方题解用了更优美的方法：用yield直接生成序列。

# yield 的作用就是把一个函数变成一个 generator
# yield from可以把可迭代对象里的每个元素一个一个的yield出来

class Solution:
    def leafSimilar(self, root1: TreeNode, root2: TreeNode) -> bool:
        def dfs(node: TreeNode):
            if not node.left and not node.right:
                yield node.val
            else:
                if node.left:
                    yield from dfs(node.left)
                if node.right:
                    yield from dfs(node.right)
        
        seq1 = list(dfs(root1)) if root1 else list()
        seq2 = list(dfs(root2)) if root2 else list()
        return seq1 == seq2