给定一个二叉树和一个值\ sum sum,请找出所有的根节点到叶子节点的节点值之和等于\ sum sum 的路径,
例如:
深度优先搜索+回溯法,最后要减去加上的节点的值,并且返回到上一届迪纳的位置。
public ArrayList<ArrayList<Integer>> pathSum (TreeNode root, int sum) {
// write code here
ArrayList<ArrayList<Integer>> res=new ArrayList<>();
ArrayList<Integer> path=new ArrayList<>();
dfs(root,sum,path,res);
return res;
}
public void dfs(TreeNode root,int sum,ArrayList<Integer> path,ArrayList<ArrayList<Integer>> res){
if(root==null){
return;
}
if(root.val==sum && root.left==null&&root.right==null){
path.add(root.val);
res.add(new ArrayList<>(path));
path.remove(path.size()-1);
}
path.add(root.val);
dfs(root.left,sum-root.val,path,res);
dfs(root.right,sum-root.val,path,res);
path.remove(path.size()-1);
}