牛客-NC8-二叉树根节点到叶子节点和为指定值的路径
NC8. 二叉树根节点到叶子节点和为指定值的路径(medium)
方法一:DFS法
思路:前序遍历的增强版,要找出根到叶节点的节点值之和等于sum的路径,所以dfs方法需要带一些标记数据,这里包括包含当前节点时的cur_sum和将当前节点纳入考虑的select列表。(注意这里有个Java常问的知识点:深浅拷贝
,我被这个东西坑了,一直得不到正确的答案,这里必须对select列表进行深拷贝,因为浅拷贝生成的列表会指向之前的列表,后续对该列表的改变都会映射到之前的列表中导致最终达到目标叶子节点会将之前遍历得到的节点全记录着。)
import java.util.*;
/* * public class TreeNode { * int val = 0; * TreeNode left = null; * TreeNode right = null; * } */
public class Solution {
/** * * @param root TreeNode类 * @param sum int整型 * @return int整型ArrayList<ArrayList<>> */
ArrayList<ArrayList<Integer>> ret = new ArrayList<>();
public ArrayList<ArrayList<Integer>> pathSum (TreeNode root, int sum) {
// write code here
if (root == null) return ret;
dfs(root, sum, new ArrayList<Integer>());
return ret;
}
public void dfs(TreeNode node, int cur_sum, ArrayList<Integer> select) {
ArrayList<Integer> newSelect = new ArrayList<>(select);
newSelect.add(node.val);
cur_sum -= node.val;
// 到达叶节点
if (node.left == null && node.right == null) {
if (cur_sum == 0) {
ret.add(newSelect);
}
}
if (node.left != null) {
dfs(node.left, cur_sum, newSelect);
}
if (node.right != null) {
dfs(node.right, cur_sum, newSelect);
}
}
}
时间复杂度: O(N),考虑最坏情况下,二叉树变成一个长链结构,此时复杂度最高。
空间复杂度: O(N),需要考虑系统递归栈,最坏情况下,全部节点需要入栈。
总结:这道题复习了浅拷贝和深拷贝,这里分享给大家一个link可以复习深浅拷贝。