题目链接: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);
}
};