leetcode 路径总和 II

最新推荐文章于 2022-02-18 17:57:13 发布

zhengqiuQAQ

最新推荐文章于 2022-02-18 17:57:13 发布

阅读量165

点赞数 2

分类专栏： LEETCODE 文章标签： c++ 二叉树 dfs

本文链接：https://blog.csdn.net/qq_22684941/article/details/120042543

版权

LEETCODE 专栏收录该内容

87 篇文章 0 订阅

订阅专栏

113. 路径总和 II

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/path-sum-ii。

使用递归DFS遍历所有路径进行判断，将符合的保存下来。
使用递归需注意在递归函数中我们需要对临时路径tmp_path删除最后一个元素，避免递归到其它分支时还保存着上一个分支的值。

/**
 * 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>> pathSum(TreeNode* root, int targetSum) {
        DFS(root, targetSum);
        return result;
    }
    void DFS(TreeNode* root, int targetSum) {
        if (root == nullptr) return;
        tmp_path.emplace_back(root->val);
        targetSum -= root->val;
        if (root->left == nullptr && root->right == nullptr && targetSum == 0) {
            result.emplace_back(tmp_path);
        }
        DFS(root->left, targetSum);
        DFS(root->right, targetSum);
        tmp_path.pop_back();
    }
private:
    vector<vector<int>> result;
    vector<int> tmp_path;
};