叶子相似的树
难度:
标签:
题目描述
请考虑一棵二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。
举个例子,如上图所示,给定一棵叶值序列为 (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,2,3], root2 = [1,3,2] 输出:false
提示:
- 给定的两棵树结点数在
[1, 200]
范围内 - 给定的两棵树上的值在
[0, 200]
范围内
代码结果
运行时间: 22 ms, 内存: 16.0 MB
/*
题目思路:
1. 使用Java Stream来获取树的叶值序列。
2. 创建一个递归方法来生成叶值序列的Stream。
3. 比较两棵树的叶值序列是否相同。
*/
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}
public class Solution {
public boolean leafSimilar(TreeNode root1, TreeNode root2) {
List<Integer> leaves1 = getLeaves(root1).collect(Collectors.toList());
List<Integer> leaves2 = getLeaves(root2).collect(Collectors.toList());
return leaves1.equals(leaves2);
}
private Stream<Integer> getLeaves(TreeNode node) {
if (node == null) {
return Stream.empty();
}
if (node.left == null && node.right == null) {
return Stream.of(node.val);
}
return Stream.concat(getLeaves(node.left), getLeaves(node.right));
}
}
解释
方法:
此题解的主要思路是通过两个辅助函数(dp1和dp2)来分别遍历两棵二叉树,并在遍历过程中收集每棵树的所有叶子节点的值。这些值被存储在两个列表a和b中。在遍历完成后,比较这两个列表。如果两个列表相同,说明两棵树的叶值序列相同,函数返回true;否则返回false。遍历树使用的是深度优先搜索(DFS)。
时间复杂度:
O(n)
空间复杂度:
O(n)
代码细节讲解
🦆
为什么选择深度优先搜索(DFS)而不是广度优先搜索(BFS)来收集叶子节点的值?
▷🦆
在比较两个叶子序列列表时,如果两棵树的叶子节点数目不相等会发生什么?
▷🦆
如果树中存在值为null的节点,这种情况如何处理?题解中是否考虑了这一点?
▷🦆
题解中对于叶子节点的判断是基于节点的左右子节点是否为null,这种方法是否适用于所有二叉树结构?
▷