题目来源:https://leetcode-cn.com/problems/path-sum-ii/description/
题目描述:
算法描述:
1.该题是上篇博客的升级版。详情请见https://blog.csdn.net/qq_39241239/article/details/82749696
2.该题比较上一题的难度在于需要输出每一条路径。所以该题应该采用回溯算法。
3.首先需要一个结果集合list用来存放最终所有路径。还需要一个中间集合arrlist用来存放一条路径。
4.每当遍历到一个结点的时候,先将该结点的值加入中间集合中。如果当前节点没有子女并且到该结点的路径符合要求,就将这个中间集合加入到结果集合当中。如果该结点有左子树,就遍历左子树。如果该结点有右子树那么就遍历右子树。
5.需要注意的是,当遍历完一条路径(不论该路径是否符合要求)回到函数调用的地方时,都要将中间集合最后一个元素删除(这里就体现出了回溯)。
代码如下:
class Solution {
public List<List<Integer>> pathSum(TreeNode root, int sum) {
//定义一个结果集合
List<List<Integer>> list = new ArrayList<>();
if (root == null) {
return list;
}
//定义一个中间集合
ArrayList<Integer> arrlist = new ArrayList<>();
path(root, list, arrlist, sum);
return list;
}
public void path(TreeNode root, List<List<Integer>> list, ArrayList<Integer> arrlist, int nowSum) {
//将当前节点的值加入中间集合中
arrlist.add(root.val);
nowSum -= root.val;
if (root.left == null && root.right == null) {
if (nowSum == 0) {
//如果当前节点无子女,就判断该路径是否符合要求,如果符合那么就将该路径加入结果集合中
list.add(new ArrayList<Integer>(arrlist));
}
//遍历完一条路径就回到调用点,这里也可以不写return,也会自动返回。但是加return可以略微提高效率。
return;
}
if (root.left != null) {
//左子树不为空就递归遍历左子树,要注意的是函数返回后,要将中间集合最后一个元素删除(这里体现了回溯)
path(root.left, list, arrlist, nowSum);
arrlist.remove(arrlist.size() - 1);
}
if (root.right != null) {
//右子树不为空就递归遍历右子树,要注意的是函数返回后,要将中间集合最后一个元素删除(这里体现了回溯)
path(root.right, list, arrlist, nowSum);
arrlist.remove(arrlist.size() - 1);
}
}
}