872. 叶子相似的树
题目链接:https://leetcode-cn.com/problems/leaf-similar-trees/
难度:简单
题解
思路:我们可以使用数组存储树对应的叶值序列,然后再对数组进行比较判断。对于数组的获取,我们可以使用深度优先搜索,将左右子节点都不存在的叶节点存入数组。
class Solution {
public boolean leafSimilar(TreeNode root1, TreeNode root2) {
//新建两数组
List<Integer> list1 = new ArrayList<>();
List<Integer> list2 = new ArrayList<>();
dfs(root1,list1);
dfs(root2,list2);
//判断数组中的是否相同
if(list1.size()!=list2.size()) return false;
for(int i=0;i<list1.size();i++){
if(list1.get(i)!=list2.get(i)) return false;
}
return true;
}
//函数功能是将是叶节点的值存入数组list
public void dfs(TreeNode root,List<Integer> list){
if(root==null) return ;
if(root.left==null&&root.right==null){
list.add(root.val);
return;
}
dfs(root.left,list);
dfs(root.right,list);
return ;
}
}
总结
对于需要递归使用的函数,要先明确函数的功能,并坚信该功能是"成功实现"的(在递归时可能还未将递归函数编写完成),再进行编码,而不应该直接跳入递归当中去。