剑指Offer二叉树中和为某一值的路径

题目描述:

输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。

思路分析:

采用先序遍历思想,当遇到第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();
}

(完)

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章