leetcode437——前缀和

Posted 2023-02-24 dtwd886

题目链接：https://leetcode-cn.com/problems/path-sum-iii/

思路：

以map来记录路径上结点的前缀和，key代表着路径上的前缀和，val代表着当前结点路径上的前缀和的个数

针对每个结点记录前缀和，并判断之前在路径上的前缀和cur_sum-target的个数，加上这个数，

并将当前结点前缀和cur_sum加入到map中，使val加一。然后递归加上左右结点产生的值为target的路径个数。

最后将当前结点前缀和对应val减一。

/**
 * 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:
    int DFS(TreeNode* node,int target,int cur_sum)
    
        if(!node)return 0;
        cur_sum+=node->val;
        int res=0;
        res+=record_prefix_sum.find(cur_sum-target)!=record_prefix_sum.end()?record_prefix_sum[cur_sum-target]:0;
        record_prefix_sum[cur_sum]=record_prefix_sum.find(cur_sum)!=record_prefix_sum.end()?
        record_prefix_sum[cur_sum]+1:1;
        res+=DFS(node->left,target,cur_sum);
        res+=DFS(node->right,target,cur_sum);
        record_prefix_sum[cur_sum]-=1;
        return res;
    
    int result=0;
    map<int,int>record_prefix_sum;
    int pathSum(TreeNode* root, int sum) 
        record_prefix_sum[0]=1;
        return DFS(root,sum,0);

    
;