题目描述:
输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
思路分析:
采用先序遍历思想,当遇到第1个结点curr,用sum-curr.val,然后将curr.val放入列表path中,当当前结点为叶子结点况且sum=0,将path加入结果集合中。否则将删除path的最后一个结点。结束本轮递归,进行下次递归。
参考代码:
private List<List<Integer>> res= new LinkedList<>();
private LinkedList<Integer> path = new LinkedList<>();
public List<List<Integer>> pathSum(TreeNode root, int sum) {
if(root == null)return res;
dfs(root, sum);
return res;
}
public void dfs(TreeNode root, int sum){
if(root == null)return;
path.add(root.val);
sum -= root.val;
if(root.left == null && root.right == null && sum == 0)
res.add(new LinkedList(path));
dfs(root.left, sum);
dfs(root.right, sum);
path.removeLast();
}
注意:
不是直接将path加入res中,因为后续还要操作path,所以重新创建一个list返回。
res.add(new LinkedList(path));
(完)