872. 叶子相似的树

合集 - LeetCode 树(11)

1.687. 最长同值路径2024-09-112.124. 二叉树中的最大路径和2024-09-113.2385. 感染二叉树需要的总时间2024-09-114.【模板题】- 144. 二叉树的前序遍历2024-09-125.【模板题】- 94. 二叉树的中序遍历2024-09-126.【模板题】- 145. 二叉树的后序遍历2024-09-12

7.872. 叶子相似的树2024-09-12

8.LCP 44. 开幕式焰火2024-09-129.404. 左叶子之和2024-09-1210.671. 二叉树中第二小的节点2024-09-1211.662. 二叉树最大宽度2024-09-24

收起

题目链接	872. 叶子相似的树
思路	通过DFS的方式得到树的叶子节点序列
题解链接	官方题解
关键点	yeild from dfs(node.left)
时间复杂度	$O(n_1+n_2)$
空间复杂度	$O(n_1+n_2)$

代码实现：

class Solution:
    def leafSimilar(self, root1: Optional[TreeNode], root2: Optional[TreeNode]) -> bool:
        if root1 is None or root2 is None:
            return root1 is root2
        
        def dfs(node):
            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)
        
        return list(dfs(root1)) == list(dfs(root2))