题号:剑指 Offer 34
题目名:二叉树中和为某一值的路径
原题URL:https://leetcode-cn.com/problems/er-cha-shu-zhong-he-wei-mou-yi-zhi-de-lu-jing-lcof/
题目描述
输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
示例
示例 1:
给定如下二叉树,以及目标和 sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
[
[5,4,11,2],
[5,8,4,5]
]
限制
节点总数 <= 10000
思路
1.使用dfs进行回溯剪枝
2.每次遍历一个结点就将值在sum基础上减
3.当sum==0的时候,就可以输出结果,将当前集合加入到返回集合中
4.注意每次进行递归的时候需要进行剪枝
解题代码
public class Solution {
List<List<Integer>> res = new ArrayList<>();
public List<List<Integer>> pathSum(TreeNode root, int sum) {
if(root==null) return res;
List<Integer> list = new ArrayList<>();
findPath(root,list,sum);
return res;
}
private void findPath(TreeNode root, List<Integer> list, int sum) {
if(root==null) return;
//将当前节点的值加入sum
list.add(root.val);
//如果当前节点是根节点
if(root.left==null&&root.right==null) {
if(sum-root.val==0) {
res.add(new ArrayList<>(list));
}
return;
}
//遍历左子树
findPath(root.left,list,sum-root.val);
//回溯
if(root.left!=null) list.remove(list.size()-1);
findPath(root.right,list,sum-root.val);
if(root.right!=null) list.remove(list.size()-1);
}
}