题目描述
请考虑一棵二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。
举个栗子,如上图,给定一棵叶值序列为 (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
思路
简单题,就是把树遍历一下,然后记录一下叶子节点,最后对比两棵树的叶子节点是不是都一样就行啦~
这里采用中序遍历,时间和空间复杂度都为O(n)。
代码
class Solution {
public boolean leafSimilar(TreeNode root1, TreeNode root2) {
List<Integer> res1 = new ArrayList<Integer>();
List<Integer> res2 = new ArrayList<Integer>();
Inorder(res1,root1);
Inorder(res2,root2);
if(res1.size() != res2.size())
return false;
for(int i = 0;i < res1.size();i++) {
if(res1.get(i) != res2.get(i))
return false;
}
return true;
}
private void Inorder(List<Integer> res,TreeNode root) {
if(root == null)
return;
Inorder(res,root.left);
if(root.left == null && root.right == null) {
res.add(root.val);
}
Inorder(res,root.right);
}
}