0%

路径总和 II


题目描述

给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。
说明: 叶子节点是指没有子节点的节点。

示例:

1
2
3
4
5
6
7
8
9
10
11
12
13
给定如下二叉树,以及目标和 sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
[
[5,4,11,2],
[5,8,4,5]
]

题解

做此题之前,可以先做一道简单版的路径总和,简单版的只要遍历全部节点,遇到满足条件的路径返回 true 即可,而此题不仅要遍历全部节点,还要记录满足条件的路径。此题是典型的回溯算法,思路就是,先声明一个LinkedList<Integer> path保存路径,List<List<Integer>> ans保存所有路径,我们每遍历到一个节点,就将这个节点的值保存在 path 中,当判断到子节点,如果此节点的值满足 sum 等于 val,则将 path 加入到 ans 中,否则继续进行递归遍历左右子树。注意入参 root 可能为空,另外递归的结束条件也包含节点为空。当递归回溯的时候,我们就删除最后加入到 path 中的节点。

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
List<List<Integer>> ans = new LinkedList<>();
LinkedList<Integer> path = new LinkedList<>();

public List<List<Integer>> pathSum(TreeNode root, int sum) {
if (root == null) {
return ans;
}
path.add(root.val);
if (root.left == null && root.right == null) {
if (sum == root.val) {
// 此处new LinkedList<>(path)新建一个链表
// 因为如果直接add(path),ans中的路径和path是相同的引用,后面操作path后,ans中的路径也将一起被修改。
ans.add(new LinkedList<>(path));
// 此处无需return,因为后续pathSum递归的时候root.left和root.right都为空,满足递归结束条件。
}
}
pathSum(root.left, sum - root.val);
pathSum(root.right, sum - root.val);
// 递归回溯的时候删除最后一个节点,此处使用LinkedList提升效率
path.removeLast();
return ans;
}

public class TreeNode {
int val;
TreeNode left;
TreeNode right;

TreeNode(int x) {
val = x;
}
}

来源

路径总和 II | 力扣(LeetCode)
路径总和 II | 题解(LeetCode)


分享精彩,留下足迹

欢迎关注我的其它发布渠道