剑指 Offer 34. 二叉树中和为某一值的路径
题目:
给你二叉树的根节点 root 和一个整数目标和 targetSum ,找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。
叶子节点 是指没有子节点的节点。
示例1:
输入:root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出:[[5,4,11,2],[5,8,4,5]]
示例2:
输入:root = [1,2,3], targetSum = 5
输出:[]
示例3:
输入:root = [1,2], targetSum = 0
输出:[]
解题思路:
本问题是典型的二叉树方案搜索问题,使用回溯法解决,其包含 先序遍历 + 路径记录 两部分。
流程如下:
路径更新: 将当前节点值 root.val 加入路径 path ;
目标值更新: target = target - root.val(即目标值 target 从 target 减至 0 );
路径记录: 当root 为叶节点 且路径和等于目标值 ,则将此路径 path 加入 list 。
先序遍历: 递归左 / 右子节点。
路径恢复: 向上回溯前,需要将当前节点从路径 path 中删除,即执行 path.removeLast() 。
代码如下:
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
* 执行结果:通过
* 执行用时:1 ms, 在所有 Java 提交中击败了100.00%的用户
* 内存消耗:38.8 MB, 在所有 Java 提交中击败了37.15%的用户
* 通过测试用例:114 / 114
*/
class Solution {
LinkedList<List<Integer>> list = new LinkedList<>();
LinkedList<Integer> path = new LinkedList<>();
public List<List<Integer>> pathSum(TreeNode root, int target) {
res(root,target);
return list;
}
void res(TreeNode root,int target){
if(root == null) return;
path.add(root.val);
target-=root.val;
if(target == 0 && root.left == null && root.right == null){
list.add(new LinkedList(path));
}
res(root.left,target);
res(root.right,target);
path.removeLast();
}
}