力扣刷题(day0037)路径总和 II

最新推荐文章于 2024-11-10 14:06:08 发布

「已注销」

最新推荐文章于 2024-11-10 14:06:08 发布

阅读量328

点赞数

文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/weixin_51600314/article/details/123850767

版权

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

/**
 * 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 {
private:
    vector<vector<int>>result;
    vector<int>path;
    //递归不需要返回值，因为要遍历整棵树
    void traversal(TreeNode* cur,int count){
        //遇到叶子节点，且值为0
        if(cur->left==NULL&&cur->right==NULL&&count==0){
            result.push_back(path);
        }
        //左
        if(cur->left){
            path.push_back(cur->left->val);
            count-=cur->left->val;
            traversal(cur->left,count);//递归
            count+=cur->left->val;//回溯
            path.pop_back();//回溯
        }
        //右
        if(cur->right){
            path.push_back(cur->right->val);
            count-=cur->right->val;
            traversal(cur->right,count);//递归
            count+=cur->right->val;//回溯
            path.pop_back();//回溯
        }
        return;
    }
public:
    vector<vector<int>> pathSum(TreeNode* root, int targetSum) {
        result.clear();
        path.clear();
        if(root==NULL)return result;
        path.push_back(root->val);//把根节点放入路径
        traversal(root,targetSum-root->val);
        return result;
    }
};