输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
示例:
给定如下二叉树,以及目标和 target = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
[
[5,4,11,2],
[5,8,4,5]
]
方法:使用dfs来做,加上回溯算法,最后一行的remove表示移除最后一个,就是回溯算法。
另外,res.add的时候,因为path是引用传递,所以要新建一个把list传进去。否则通过path.remove全都移除了,最后res是个空。
class Solution {
ArrayList<List<Integer>> res = new ArrayList<>();
ArrayList<Integer> path = new ArrayList<>();
public List<List<Integer>> pathSum(TreeNode root, int sum) {
dfs(root, sum);
return res;
}
public void dfs(TreeNode root, int sum){
if(root==null){
return;
}
path.add(root.val);
int target = sum - root.val;
if(target==0 && root.left==null && root.right==null){
res.add(new ArrayList(path));
}
dfs(root.left, target);
dfs(root.right, target);
path.remove(path.size()-1);
}
}