Given a binary tree and a sum, find all root-to-leaf paths where each path's sum equals the given sum.
For example:
Given the below binary tree and sum = 22,
5
/
4 8
/ /
11 13 4
/ \ /
7 2 5 1
return
[
[5,4,11,2],
[5,8,4,5]
]
这题我一拿到就感觉应该用DFS保护现场那套来做，但没写出来，因为不知道remove在哪操作。
参考了code ganker的代码调试了一阵子AC了。
public class Solution {
List<List<Integer>> res = new ArrayList<>();
public List<List<Integer>> pathSum(TreeNode root, int sum) {
if (root == null) return res;
List<Integer> cell = new ArrayList<>();
cell.add(root.val);
helper(root, sum - root.val, cell);
return res;
}
private void helper(TreeNode root, int sum, List<Integer> cell) {
if (root == null) return;
if (root.left == null && root.right == null && 0 == sum) {
res.add(new ArrayList<>(cell));
//相当于else
return;
}
if (root.left != null ) {
cell.add(root.left.val);
helper(root.left, sum - root.left.val, cell);
cell.remove(cell.size() - 1);
}
if (root.right != null ) {
cell.add(root.right.val);
helper(root.right, sum - root.right.val, cell);
cell.remove(cell.size() - 1);
}
}
}
需要注意的地方：
- 这题跟上题不一样，它先把root加进初始cell表里去，然后才去递归找left和right。为什么不能直接把root放进去，我想是因为不好判断，因为
if (root.left != null && sum > 0) {
cell.add(root.left.val);
helper(root.left, sum - root.left.val);
cell.remove(cell.size() - 1);
}
这三行是套路，不能在套路外面add node啊。
第二是要在root为null的时候和满足条件之后就return(废话)。
第三是
res.add(new ArrayList<>(cell));这句话不能用res.add(cell); cell = new ArrayList<>();这两句来代替，因为这是dfs啊，后面还要保护现场呢，你找到一个结果之后它要在你原先的结果后面remove的，如果new了，后面remove的时候直接就out of bounds -1了。
