题目描述
输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
示例: 给定如下二叉树,以及目标和 target = 22,5 / \ 4 8 / / \ 11 13 4 / \ / \ 7 2 5 1
返回:
[ [5,4,11,2], [5,8,4,5] ]提示:
节点总数 <= 10000
方法一(回溯)
1.解题思路
用一个二维list存储最终所有的路径,用一个一维list存储每一次恰好为目标值的路径,当树中没有节点时,递归终止。递归推进过程中,需要将当前节点添加到path,同时target减去当前节点值,如果target为0表示装入path的几个值的和恰好为target,此时当前节点为叶子节点,则找到了一条路径,直接加到结果list。然后向左子节点和右子节点不断递归,同时要注意,每往左或右走一步,都要进行一次回溯(删掉path最后一个元素),相当于不断试错的过程。
2.代码实现
class Solution {
List<List<Integer>> res=new ArrayList<>();
List<Integer> path=new ArrayList<>();
public List<List<Integer>> pathSum(TreeNode root, int target) {
if(root==null) return res;
path.add(root.val);
target-=root.val;
if(target==0&&root.left==null&&root.right==null){
res.add(new ArrayList<>(path));
}
pathSum(root.left,target);
pathSum(root.right,target);
path.remove(path.size()-1);
return res;
}
}
3.复杂度分析
- 时间复杂度:需要遍历树中所有节点,所以时间复杂度为O(n)
- 空间复杂度:最坏情况下(退化为链表),递归栈的深度为n,所以空间复杂度为O(n)
剑指offer全集入口: 请戳这里