面试题34. 二叉树中和为某一值的路径
难度中等27
输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
示例:
给定如下二叉树,以及目标和 sum = 22,
5 / \ 4 8 / / \ 11 13 4 / \ / \ 7 2 5 1
返回:
[ [5,4,11,2], [5,8,4,5] ]
//回溯
// 递归解决 终止条件节点为null 返回
// 当target=0 并且左右子节点都为null的时候 进行数据添加。
// 否则的话 直接递归左右子节点。
List<List> ret = new ArrayList<>();
public List<List<Integer>> pathSum(TreeNode root, int sum) {
pathSum(root,sum,new ArrayList());
return ret;
}
public void pathSum(TreeNode root,int target,List<Integer> path){
if(root == null){
return;
}
path.add(root.val);
target -= root.val;
if(target == 0 && root.left == null && root.right == null){
ret.add(new ArrayList<>(path));
}else{
pathSum(root.left,target,path);
pathSum(root.right,target,path);
}
//回溯,状态重置,只添加不删除就是错的路径了,因为整个递归过程用的是同一个收集器path
path.remove(path.size()-1);
}