872.叶子相似的树
https://leetcode-cn.com/problems/leaf-similar-trees/
难度:简单
题目:
请考虑一棵二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。
网络异常,图片无法展示
|
举个例子,如上图所示,给定一棵叶值序列为(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 示例 3: 输入:root1 = [1], root2 = [2] 输出:false 示例 4: 输入:root1 = [1,2], root2 = [2,2] 输出:true 示例 5: 输入:root1 = [1,2,3], root2 = [1,3,2] 输出:false
分析
二叉树搜索叶子结点,要求我们对深度优先的方法具备相关掌握即可完成解题。
在深度优先搜索的过程中,我们总是先搜索当前节点的左子节点,再搜索当前节点的右子节点。
这里由于要搜索两个二叉树的叶子结点,所以单独抽离出dfs方法,进行调用后判断结果
解题:
# 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 dfs(root, ret): if not root.left and not root.right: ret.append(root.val) return if root.left: dfs(root.left, ret) if root.right: dfs(root.right, ret) ret1 = [] dfs(root1, ret1) ret2 = [] dfs(root2, ret2) return ret1 == ret2