LeetCode每日一题3 - 872.叶子相似的树

题目名称：872.叶子相似的树

难度 ：简单

题目描述

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

举个例子，如上图所示，给定一棵叶值序列为 (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,null,null,null,null,9,8]
输出：true

示例2

输入：root1 = [1], root2 = [1]
输出：true

提示

• 给定的两棵树可能会有 1 到 200 个结点。
• 给定的两棵树上的值介于 0 到 200 之间。

题目解析

思路：使用深度优先搜索的方法得到一棵树的「叶值序列」。

具体地，在深度优先搜索的过程中，我们总是先搜索当前节点的左子节点，再搜索当前节点的右子节点。如果我们搜索到一个叶节点，就将它的值放入序列中。

在得到了两棵树分别的「叶值序列」后，我们比较它们是否相等即可。

# 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 leaves(root):
            if root.left is None and root.right is None:
                yield root.val
            else:
                if root.left:
                    yield from leaves(root.left)
                if root.right:
                    yield from leaves(root.right)
        
        ls1 = list(leaves(root1)) if root1 else list()
        ls2 = list(leaves(root2)) if root2 else list()

        return ls1 == ls2

yield from后面加上可迭代对象，可以把可迭代对象里的每个元素一个一个的yield出来。

复杂度分析

• 时间复杂度：O(n_1 + n_2)，其中 n_1和 n_2分别是两棵树的节点个数。

• 空间复杂度：O(n_1 + n_2)，空间复杂度主要取决于存储「叶值序列」的空间以及深度优先搜索的过程中需要使用的栈空间。

提交结果

个人总结

到目前为止，递归这个概念对我来说依旧熟悉而陌生，有时候觉得懂了，敲代码的时候却不能得心应手，想彻底搞懂它，又觉得好陌生。对它最深刻的理解是递归的过程和数学归纳法很像。多说无益，还是要多去题目中，才能真正理解和掌握递归的使用！