题目
输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
示例:
给定如下二叉树,以及目标和 sum = 22
,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
[
[5,4,11,2],
[5,8,4,5]
]
提示:
节点总数 <= 10000
解题思路
主要思路:通过先序遍历,记录从根节点到当前节点的路径。当遍历到叶节点且路径上各节点值的和等于目标值 sum 时,将此路径加入结果列表。
递归步骤:
- 1、路径更新: 更新当前路径和,并将当前节点值 root.val 加入路径 path ;;
- 2、添加路径至结果列表:若 1)root 为叶节点;2)当前路径和等于目标值。两个条件同时满足,则将此路径 path 加入 res。
- 3、递归遍历子节点: 如果左、右节点非空,则递归遍历左、右子节点。
- 4、回溯:返回父节点时,需要将当前节点从路径 path 中删除。
复杂度分析:
时间复杂度:O(N)。N 为二叉树的节点数,先序遍历需要遍历所有节点。
空间复杂度:O(N)。最差情况下,即树退化为链表时,path 存储所有树节点,使用 O(N) 额外空间。
注:
1)记录路径时若直接执行 res.add(path) ,则是将 path 对象加入了 res。后续 path 改变时, res 中的 path 对象也会随之改变。
2)正确做法:res.add(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 {
private List<List<Integer>> res = new ArrayList<>();
public List<List<Integer>> pathSum(TreeNode root, int sum) {
if(root == null){
return res;
}
LinkedList<Integer> path = new LinkedList<>();
dfs(root, path, 0, sum);
return res;
}
private void dfs(TreeNode root, LinkedList<Integer> path, int curSum, int sum){
curSum += root.val;
path.offerLast(root.val);
// 如果是叶节点,并且路径和等于期望值,就加入结果列表
if(curSum == sum && root.left == null && root.right == null){
res.add(new LinkedList(path));
// res.add(path);
}
if(root.left != null){
dfs(root.left, path, curSum, sum);
}
if(root.right != null){
dfs(root.right, path, curSum, sum);
}
// 回溯:返回父节点时,在路径上删除当前节点
path.removeLast();
}
}