5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
[
[5,4,11,2],
[5,8,4,5]
]
值得注意的是,记录路径时若直接执行 res.add(path) ,则是将 path 对象加入了 res ;后续 path 改变时, res 中的 path 对象也会随之改变。
正确做法:res.append(new LinkedList(path)) ,相当于复制了一个 path 并加入到 res 。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
// 结果列表
LinkedList<List<Integer>> res = new LinkedList<>();
// 路径
LinkedList<Integer> path = new LinkedList<>();
public List<List<Integer>> pathSum(TreeNode root, int sum) {
recursion(root, sum);
return res;
}
public void recursion(TreeNode root, int sum) {
// 判空
if (root == null)
return;
// 首先把根结点加入到路径中
path.add(root.val);
// 路径和减去根结点的值
sum -= root.val;
// 此时需要判断一下路径和是否等于 0 且根结点不存在左右结点,若成立,把此时的路径添加到结果中
if (sum == 0 && root.left == null & root.right == null)
res.add(new LinkedList(path)); //别写成path了 path是会变的
// 递归左右子树
recursion(root.left, sum);
recursion(root.right, sum);
// 回溯,删去此时路径中的最后一个结点
path.removeLast();
}
}
// 不用LinkedList:
class Solution {
// LinkedList方便回溯
List<List<Integer>> res = new ArrayList<>();
List<Integer> path = new ArrayList<>();
public List<List<Integer>> pathSum(TreeNode root, int sum) {
recursion(root, sum);
return res;
}
public void recursion(TreeNode root, int sum) {
if (root == null)
return;
path.add(root.val);
sum -= root.val;
if (sum == 0 && root.left == null && root.right == null)
res.add(new ArrayList(path));
recursion(root.left, sum);
recursion(root.right, sum);
path.remove(path.size() - 1);
}
}