题目
给你二叉树的根节点 root
和一个整数目标和 targetSum
,找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。
叶子节点 是指没有子节点的节点。
示例 1:
输入:root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出:[[5,4,11,2],[5,8,4,5]]
1
2
2
示例 2:
输入:root = [1,2,3], targetSum = 5
输出:[]
1
2
2
示例 3:
输入:root = [1,2], targetSum = 0
输出:[]
1
2
2
提示:
- 树中节点总数在范围
[0, 5000]
内 -1000 <= Node.val <= 1000
-1000 <= targetSum <= 1000
题解
java
public List<List<Integer>> pathSum(TreeNode root, int sum) {
List<List<Integer>> result = new ArrayList<>();
// 递归求解
TrConsumer<TreeNode, List<Integer>, Integer> recursion = new TrConsumer<TreeNode, List<Integer>, Integer>() {
@Override
public void accept(TreeNode node, List<Integer> list, Integer s) {
if (node == null) {
return;
}
// 若是叶子节点
if (node.left == null && node.right == null) {
// 满足路径和
if (s == node.val) {
list.add(node.val);
result.add(list);
}
} else {
list.add(node.val);
// 递归累减左右树
this.accept(node.left, new ArrayList<>(list), s - node.val);
this.accept(node.right, new ArrayList<>(list), s - node.val);
}
}
};
recursion.accept(root, new ArrayList<>(), sum);
return result;
}
private interface TrConsumer<T, U, P> {
void accept(T t, U u, P p);
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34