请考虑一棵二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。
举个例子,如上图所示,给定一棵叶值序列为 (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], root2 = [2,2]
输出:true
示例 3:
输入:root1 = [1,2,3], root2 = [1,3,2]
输出:false
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/leaf-similar-trees
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题目比较简单,但是有一个在树遍历中的常见方式(利用栈模拟递归过程)值得了解和使用。
递归:
class Solution {
public boolean leafSimilar(TreeNode root1, TreeNode root2) {
List<Integer> l1 = new ArrayList<Integer>();
List<Integer> l2 = new ArrayList<Integer>();
func(l1, root1);
func(l2, root2);
if(l1.size() == l2.size()){
for(int i = 0; i < l1.size(); i++){
if(l1.get(i) != l2.get(i)){
return false;
}
}
return true;
}
return false;
}
void func(List l,TreeNode root){
if(root.left == null && root.right == null){
l.add(root.val);
return;
}else{
if(root.left != null){
func(l,root.left);
}
if(root.right != null){
func(l,root.right);
}
}
}
}
迭代:
class Solution {
public boolean leafSimilar(TreeNode root1, TreeNode root2) {
List<Integer> l1 = new ArrayList<Integer>();
List<Integer> l2 = new ArrayList<Integer>();
func(l1, root1);
func(l2, root2);
if(l1.size() == l2.size()){
for(int i = 0; i < l1.size(); i++){
if(l1.get(i) != l2.get(i)){
return false;
}
}
return true;
}
return false;
}
void func(List l,TreeNode root){
Deque<TreeNode> d = new ArrayDeque<TreeNode>();
while(root != null || !d.isEmpty()){
while(root != null){
d.addLast(root);
root = root.left;
}
root = d.pollLast();
if(root.left == null && root.right == null){
l.add(root.val);
}
root = root.right;
}
}
}