[LeetCode]剑指 Offer 34. 二叉树中和为某一值的路径

Posted 2022-09-12 Spring-_-Bear

给你二叉树的根节点 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
输出：[]

提示：

• 树中节点总数在范围 [0, 5000] 内
• -1000 <= Node.val <= 1000
• -1000 <= targetSum <= 1000

题解：

算法流程：

pathSum(root, sum) 函数：

• 初始化： 结果列表 res ，路径列表 path
• 返回值： 返回 res 即可

recur(root, tar) 函数：

• 递推参数： 当前节点 root ，当前目标值 tar
• 终止条件： 若节点 root 为空，则直接返回
• 递推工作：
  1. 路径更新： 将当前节点值 root.val 加入路径 path
  2. 目标值更新： tar = tar - root.val（即目标值 tar 从 sum 减至 0)
  3. 路径记录： 当 root 为叶节点且路径和等于目标值 ，则将此路径 path 加入 res
  4. 先序遍历： 递归左 / 右子节点
  5. 路径恢复： 向上回溯前，需要将当前节点从路径 path 中删除，即执行 path.pop()

class Solution 
    private List<List<Integer>> res = new LinkedList<>();
    private LinkedList<Integer> path = new LinkedList<>();
    
    /**
     * 剑指 Offer 34. 二叉树中和为某一值的路径
     */
    public List<List<Integer>> pathSum(TreeNode root, int target) 
        recur(root, target);
        return res;
    

    private void recur(TreeNode root, int target) 
        if (root == null) 
            return;
        
        path.add(root.val);
        target -= root.val;

        // 路径和为 target 且当前结点为叶子结点
        if (target == 0 && root.left == null && root.right == null) 
            res.add(new LinkedList(path));
        

        recur(root.left, target);
        recur(root.right, target);
        // 向上回溯前，需要将当前节点值从路径 path 中删除
        path.removeLast();
    

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/er-cha-shu-zhong-he-wei-mou-yi-zhi-de-lu-jing-lcof