题干:
给你二叉树的根节点 root 和一个整数目标和 targetSum ,找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。
叶子节点 是指没有子节点的节点。
输入:root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出:[[5,4,11,2],[5,8,4,5]]
class Solution {
List<List<Integer>> ans = new ArrayList<>();
LinkedList<Integer> path = new LinkedList<>();
public List<List<Integer>> pathSum(TreeNode root, int target) {
recur(root,target);
return ans;
}
public void recur(TreeNode root,int target){
if(root==null) return;
path.add(root.val);
target-=root.val;
if(target==0&&root.left==null&&root.right==null) ans.add(new LinkedList(path));
recur(root.left,target);
recur(root.right,target);
path.removeLast();
}
}
path是单条路径的list,这里要用LinkedList是因为在递归的时候,如果没有达到target=0,那么就需要删除这个节点,而ArrayList在删除这方面的效率没有LinkedList高。
这里的递归只有两个要素,即终止条件和本轮递归所要进行的操作:
终止条件:该节点不存在,也就是递归到了叶子节点的下方时,终止递归。
本轮递归的操作:首先把当前经过的节点加入path中,然后判断target在减去当前节点的值后是否为0且此时是否为叶子节点,如果同时满足,就把当前path加入到ans中。之后再去遍历当前节点的后续节点。当不满足条件时,删除掉最后加入的节点(此时删除的节点既可能是非叶子节点,也可能是叶子节点)。
PS:回溯