思路:使用回溯
class Solution {
private List<List<Integer>> res;
// 二叉树中和为某一值的路径
public List<List<Integer>> pathSum(TreeNode root, int sum) {
res = new ArrayList<>();
backtrack(root, sum, new ArrayList<>());
return res;
}
private void backtrack(TreeNode node, int target, List<Integer> collector) {
if (node == null) {
return;
}
collector.add(node.val);
target -= node.val;
if (target == 0 && node.left == null && node.right == null) {
res.add(new ArrayList<>(collector));
} else {
backtrack(node.left, target, collector);
backtrack(node.right, target, collector);
}
collector.remove(collector.size() - 1);
}
}
需要注意的几点:
1.在最后判定成功 时,为什么不能直接res.add(collector);return;
因为 回溯到上一层之前没做状态重置,整个递归过程用的一个collector 需要保存他的当前状态,要是保存collector 最后存进去的全是空的。
2.为什么collector进行回溯,但是target没有回溯?
因为每次用的target都是不一样的,值传递,只要是没有在下层回溯前修改target的值(如target -= root.val),就不用还原target,因为每次会传递一个target的拷贝进入下层回溯!
这个题目很经典,不止碰到过一次,需要反复训练。