这题一开始回溯错地方了,导致花时有点长
首先对题目分析,找出路径中等于targetSum的,那么有三种情况,一种是root->val 刚刚好等于targetSum,另一种则是路径往下,还有一种是出现的节点在后面。第一和第二种可以结合成一个函数。边看边说吧~
class Solution {
public:
int dfs(TreeNode *root,int targetSum)
{
if(root==nullptr)
return 0;
int count=0;
targetSum -=root->val;//判断这个节点是否等于targetSum
if(targetSum == 0)
++count; //等于++
return count+dfs(root->left,targetSum)+dfs(root->right,targetSum);//从当前节点为头,往下判断次数
}
int pathSum(TreeNode* root, int targetSum) {
if(root == nullptr) //如果等于nullptr,则说明没有这种方法
return 0;
else
return dfs(root,targetSum)+pathSum(root->left,targetSum)+pathSum(root->right,targetSum); //dfs方法是判断当前节点开始的情况,而重新调用pathSum方法是表示不从这个节点开始,从下个节点开始。所以targetSum直接用就好(继承过来)
}
};