437. 路径总和 III
题目:给定一个二叉树,它的每个结点都存放着一个整数值。找出路径和等于给定数值的路径总数。路径不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的(只能从父节点到子节点)。
二叉树不超过1000个节点,且节点数值范围是 [-1000000,1000000] 的整数。
class Solution {
public:
int ans = 0;
//以每个节点为根节点DFS遍历
int pathSum(TreeNode* root, int sum) {
if(root == nullptr) return ans;
dfs(root, sum);
pathSum(root->left, sum);
pathSum(root->right, sum);
return ans;
}
//前序递归遍历每个节点
void dfs(TreeNode* root, int sum)
{
if(root == nullptr) return;
sum -= root->val;
if(sum == 0) ans++;
dfs(root->left, sum);
dfs(root->right, sum);
}
};