请考虑一颗二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。
举个例子,如上图所示,给定一颗叶值序列为 (6, 7, 4, 9, 8)
的树。
如果有两颗二叉树的叶值序列是相同,那么我们就认为它们是 叶相似 的。
如果给定的两个头结点分别为 root1
和 root2
的树是叶相似的,则返回 true
;否则返回 false
。
提示:
- 给定的两颗树可能会有
1
到100
个结点。
用两个list,先将两个数的叶值序列求出来,再判断这两个序列是否相等即可。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
List<Integer> l1,l2;
public boolean leafSimilar(TreeNode root1, TreeNode root2) {
l1 = new ArrayList<>();
l2 = new ArrayList<>();
getLeaves(root1,1);
getLeaves(root2,2);
if(l1.size()!=l2.size())return false;
int n = l1.size();
for(int i=0;i<n;i++){
if(l1.get(i)!=l2.get(i))return false;
}
return true;
}
void getLeaves(TreeNode r,int i){
if(r==null)return;
if(r.left==null&&r.right==null){
//i==1?l1.add(r.val):l2.add(r.val);
if(i==1){
l1.add(r.val);
}
else{
l2.add(r.val);
}
return;
}
if(r.left!=null){
getLeaves(r.left,i);
}
if(r.right!=null){
getLeaves(r.right,i);
}
}
}