[leetCode]113. 路径总和 II
题目
给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
[
[5,4,11,2],
[5,8,4,5]
]
递归
- 确定递归函数返回值以及函数参数
由于要遍历整一棵树所以返回值为void,参数为(TreeNode root, int sum)
变量sum用以记录当前节点对应的值 - 确定退出条件
如果是叶子节点并且sum等于叶子节点的值则添加当前路径到答案 - 确定单层逻辑
要注意回溯
class Solution {
List<List<Integer>> ans = new ArrayList<>();
public List<List<Integer>> pathSum(TreeNode root, int sum) {
paths(root, sum, new ArrayList<>());
return ans;
}
private void paths(TreeNode root, int sum, List<Integer> path) {
if (root == null) return;
path.add(root.val);
if (root.left == null && root.right == null && root.val == sum) {
ans.add(new ArrayList<>(path));
path.remove(path.size() - 1);
return;
}
paths(root.left, sum - root.val, path);
paths(root.right, sum - root.val, path);
path.remove(path.size() - 1);
}
}