？872. 叶子相似的树（Python）

题目

难度：★★☆☆☆
类型：二叉树

请考虑一颗二叉树上所有的叶子，这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。

叶值序列

举个例子，如上图所示，给定一颗叶值序列为 (6, 7, 4, 9, 8) 的树。

如果有两颗二叉树的叶值序列是相同，那么我们就认为它们是 叶相似 的。

如果给定的两个头结点分别为 root1 和 root2 的树是叶相似的，则返回 true；否则返回 false 。

提示

给定的两颗树可能会有 1 到 100 个结点。

解答

这里参考官方解答:

首先，让我们找出给定的两个树的叶值序列。之后，我们可以比较它们，看看它们是否相等。要找出树的叶值序列，我们可以使用深度优先搜索。如果结点是叶子，那么 dfs 函数会写入结点的值，然后递归地探索每个子结点。这可以保证按从左到右的顺序访问每片叶子，因为在右孩子结点之前完全探索了左孩子结点。

class Solution:
    def leafSimilar(self, root1, root2):
        """

        :param root1: TreeNode
        :param root2: TreeNode
        :return: bool
        """

        def dfs(node):
            if node:
                if not node.left and not node.right:
                    yield node.val
                yield from dfs(node.left)
                yield from dfs(node.right)

        return list(dfs(root1)) == list(dfs(root2))

如有疑问或建议，欢迎评论区留言~