目录

0872:叶子相似的树(1287 分)

力扣第 94 场周赛第 1 题

题目

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

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

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

如果给定的两个根结点分别为 root1root2 的树是叶相似的,则返回 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,2,3], root2 = [1,3,2]
输出:false

提示:

  • 给定的两棵树结点数在 [1, 200] 范围内
  • 给定的两棵树上的值在 [0, 200] 范围内

分析

前序遍历中保存叶子的值即得到叶值序列。

解答

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
def leafSimilar(self, root1: TreeNode, root2: TreeNode) -> bool:
    def leaf(root):
        res, stack = [], [root]
        while stack:
            node = stack.pop()
            if node:
                if not node.left and not node.right:
                    res.append(node.val)
                stack.extend([node.right, node.left])
        return res
    return leaf(root1) == leaf(root2)

48 ms