1、题目
给定一个二叉树,它的每个结点都存放着一个整数值。
找出路径和等于给定数值的路径总数。
路径不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的(只能从父节点到子节点)。
二叉树不超过1000个节点,且节点数值范围是 [-1000000,1000000] 的整数
链接:https://leetcode-cn.com/problems/path-sum-iii/
2、思路分析
双重递归:先开始调用dfs在根节点处查找是否有满足目标值的路径,用count记录路径的满足条件的总和;在调用递归函数pathSum遍历根节点的左子树和右子树的每一个节点,算出每个节点的满足条件的个数。
/**
* 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 count = 0;
int pathSum(TreeNode* root, int targetSum) {
if(!root)
return 0;
dfs(root, targetSum);
pathSum(root->left, targetSum);
pathSum(root->right, targetSum);
return count;
}
void dfs(TreeNode* root, int sum)
{
if(!root)
return;
sum -= root->val;
if(sum == 0)
count++;
dfs(root->left, sum);
dfs(root->right, sum);
}
};