这题是112. 路径总和 的变体,需要注意两个题的区别,112题是求二叉树是否存在一个与目标值相等的和,如果有相等的值,直接返回,不用管其他分支是否还有,因为这个树上已经存在一个与目标值相等的分支,因此这题不用遍历整个树;而本题是需要找到所有与目标值相等的分支,因此需要对整个树遍历,当遍历到符合目标值的时候,就将这个分支记录下来,直到整棵树遍历后再去看有多少结果,最终输出所有符合要求的树分支。
/**
* 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 {
//这里用了List,也可以用Deque,Deque可以用来做队列和栈使用
List<List<Integer>> res = new LinkedList<>();
Deque<Integer> path = new LinkedList<Integer>();
public List<List<Integer>> pathSum(TreeNode root, int targetSum) {
dfs(root,targetSum);
return res;
}
private void dfs(TreeNode root, int targetSum) {
if (root == null) return;
//这里不管该节点是否最后符合结果,都加入path,如果最终targetSum不符合,那么不加入res即可
path.offer(root.val);
targetSum -= root.val;
if (root.left == null && root.right == null && targetSum == 0) {
res.add(new LinkedList<Integer>(path));
}
dfs(root.left, targetSum);
dfs(root.right, targetSum);
//需要删除当前path中最后一个值,因为经过上面遍历后这个值已经不符合结果了,
//需要回溯从上一个值的右节点继续找
path.pollLast();
}
}
这里有一个问题,最后回溯的那一步:path.pollLast(); 为什么dfs递归了左右子树,这里只回溯了一次?
因为这里回溯的是当前节点,打断点可以看到前面的dfs已经把子树的值移出了。只有遍历到叶子节点的时候才会把最后加入的叶子节点弹出栈。已经使用,没有其他价值。只有是叶子节点的时候才需要回溯,左右子树只有一个为空返回的话是不需要回溯的。
pop是pop的 path.offer(root.val);
这里的poll,其实是移出的当前的节点,而移出当前节点的前提是已经遍历了当前节点的左右子树,这里的dfs已经遍历了当前节点的左右子树,所以这里不矛盾。