题目:
给你二叉树的根节点 root 和一个整数目标和 targetSum ,找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。
叶子节点 是指没有子节点的节点。
解答:
思路:
- 这道题的思路就是遍历所有从根节点到叶子节点的路径,然后判断路径上的数字和是否等于target。
- 难点:如何保留路径上的数字?
- 在dfs函数中添加一个ArrayList参数,每次遍历一个节点就将节点的值添加进去。在后面如果得知这个节点不是满足要求的路径,就使用remove函数将这个节点的值移除。
- 细节:遍历到叶子节点时,如果发现这条路径是满足条件的,这时需要将参数中的ArrayList添加进返回值res中,但是不能直接通过res.add(该ArrayList),因为这个参数中的ArrayList是所有路径公有的,如果添加进去了,后面其他路径一修改,res中也会发生修改,所以这时需要new一个新的ArrayList,再添加进res中。
代码:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
private List<List<Integer>> res;
public List<List<Integer>> pathSum(TreeNode root, int target) {
res=new ArrayList<>();
dfs(root,target,new ArrayList<>());
return res;
}
private void dfs(TreeNode node,int sum,ArrayList<Integer> collector){
if(node==null) return;
collector.add(node.val);
sum-=node.val;
if(sum==0&&node.left==null&&node.right==null){
res.add(new ArrayList<>(collector));
}else{
dfs(node.left,sum,collector);
dfs(node.right,sum,collector);
}
collector.remove(collector.size()-1);
}
}