请考虑一颗二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列。
举个例子,如上图所示,给定一颗叶值序列为 (6, 7, 4, 9, 8)
的树。
如果有两颗二叉树的叶值序列是相同,那么我们就认为它们是 叶相似 的。
如果给定的两个头结点分别为 root1
和 root2
的树是叶相似的,则返回 true
;否则返回 false
。
样例输入:
[3,5,1,6,2,9,8,null,null,7,4]
[3,5,1,6,7,4,2,null,null,null,null,null,null,9,8]
样例输出:
true
提示:
- 给定的两颗树可能会有
1
到100
个结点。
DFS
写个python
玩,感觉函数(是叫函数吗滑稽)定义位置好诡异
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def leafSimilar(self, root1, root2):
"""
:type root1: TreeNode
:type root2: TreeNode
:rtype: bool
"""
ans1, ans2 = list(), list()
def dfs(root, ans):
if not root.left and not root.right:
ans.append(root.val)
return
if root.left:
dfs(root.left, ans)
if root.right:
dfs(root.right,ans)
dfs(root1, ans1)
dfs(root2, ans2)
return ans1 == ans2