两种方法都是参考了题解,看完自己写一遍弄明白,今日状态极差(我是FW)有时间会回头做这一系列前面两道题 打卡!
第一种解法:穷举+dfs
class Solution
{
public:
int rootSum(TreeNode *root, int targetSum)//对于某一起点dfs寻找其中是否存在满足条件的路径
{
if (!root)
return 0;
int res = 0;
if (root->val == targetSum)
{
res = 1;//不可以在这里直接返回因为后面可能有+n-n的情况存在
}
res += rootSum(root->left, targetSum - root->val);
res += rootSum(root->right, targetSum - root->val);
return res;
}
int pathSum(TreeNode *root, int targetSum)//对于整个树dfs遍历每一个点,最每一个点进行判别
{
if (!root)
return 0;
int res = 0;
res += rootSum(root, targetSum);
res += pathSum(root->left, targetSum);
res += pathSum(root->right, targetSum);
return res;
}
};
第二种解法:前缀和+dfs
class Solution
{
public:
map<long long, int> prefix;
int Res(TreeNode *root, int curr, int targrtSum)
{
if (!root)
return 0;
int res = 0;
curr += root->val;
if (prefix.count(curr - targrtSum))
{
res += prefix[curr - targrtSum];
}
prefix[curr]++; //放在这里而不能放在if前是因为避免targetSum为零时出错!
res += Res(root->left, curr, targrtSum);
res += Res(root->right, curr, targrtSum);
prefix[curr]--;//退出当前节点一定记得更新map删除相应的前缀和
return res;
}
int pathSum(TreeNode *root, int targetSum)
{
prefix[0] = 1;//对应前缀和刚好等于targetSum时计次
return Res(root, 0, targetSum);
}
};