【前端面试常见算法题系列】113. 路径总和 II（中等）

“Offer 驾到，掘友接招！我正在参与2022春招打卡活动，点击查看活动详情。”

一、题目描述

给你二叉树的根节点 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

二、思路分析

这道题是 【前端面试常见算法题系列】112. 路径总和（简单） 的进阶版，要求找到满足题意的所有路径并返回各自路径的所有节点值。

思路如下：

• 每次遍历时将当前节点值存进 arr 数组。
• 找到符合条件的路径就直接将 arr 数组 push 进答案数组中；
• 若不符合条件且未在当前路径遍历完，则继续遍历、存储当前节点值；
• 若不符合条件且在当前路径遍历完，则将当前节点值从 arr 数组中取出，更换一条路径，push 新路径的当前节点值，继续判断、遍历等。

三、AC 代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> res;
    vector<vector<int>> pathSum(TreeNode* root, int targetSum) {
        if (!root) return res;
        vector<int> arr;
        traverse(root, targetSum, arr);
        return res;
    }
    void traverse(TreeNode* root, int targetSum, vector<int>arr) {
        if (!root) return;

        targetSum -= root -> val;
        if (!root -> left && !root -> right) {
            if (targetSum == 0) {
                arr.push_back(root -> val);
                res.push_back(arr);
            }
        }
        arr.push_back(root -> val);
        traverse(root -> left, targetSum, arr);
        arr.pop_back();

        arr.push_back(root -> val);
        traverse(root -> right, targetSum, arr);
        arr.pop_back();
    }
};

四、总结

进阶题是在基础题的基础上增大难度，但解法上有相似之处，只是对数据的处理不同而已，多思考才能让一道题举一反三，下次遇到同类型的题时也不会没有思路了。