请考虑一棵二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。
举个例子,如上图所示,给定一棵叶值序列为 (6, 7, 4, 9, 8) 的树。
如果有两棵二叉树的叶值序列是相同,那么我们就认为它们是 叶相似 的。
如果给定的两个根结点分别为 root1 和 root2 的树是叶相似的,则返回 true;否则返回 false 。
这道题只是个二叉树遍历问题,我们需要定义一个函数,输入二叉树返回它的叶值序列,然后调用函数得到两棵二叉树的叶值序列,比较是否相同。
# 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:
leaves1, leaves2 = [], []
def getleaf(root, leaves):
if not root:
return
if root.left==None and root.right==None: # 判断是否为叶子
leaves.append(root.val)
return
getleaf(root.left, leaves)
getleaf(root.right, leaves)
getleaf(root1, leaves1)
getleaf(root2, leaves2)
return leaves1 == leaves2