872. 叶子相似的树

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

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

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

如果给定的两个头结点分别为root1root2的树是叶相似的,则返回 true;否则返回 false 。

提示:

给定的两颗树可能会有 1100 个结点。

思路:

  • 通过leaf函数递归获取树的叶子节点序列,比较叶子节点序列是否相等即可
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def leafSimilar(self, root1, root2):
        """
        :type root1: TreeNode
        :type root2: TreeNode
        :rtype: bool
        """
        return self.leaf(root1) == self.leaf(root2)

    def leaf(self, root):
        if not root:
            return []
        if not root.left and not root.right:
            return [root.val]
        
        return self.leaf(root.left) + self.leaf(root.right)
posted @ 2018-09-21 09:46  yuyin  阅读(78)  评论(0编辑  收藏  举报