1. 问题描述
2. 样例说明
3. 解法一:两次DFS
算法思想
使用两次dfs
解决
- 第一次
dfs
遍历二叉树每个结点。 - 第二次
dfs
以每个结点为根进行遍历,寻找等于目标值的路径个数。
代码
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* };
*/
class Solution {
public:
int ans=0; //全局变量记录路径
void dfs(TreeNode* root, int sum){ //以某结点为根的路径数
if(!root) return;
sum-=root->val;
if(sum==0) ans++; //满足条件
dfs(root->left,sum); //递归
dfs(root->right,sum);
}
int FindPath(TreeNode* root, int sum) {
if(!root) return 0;
dfs(root,sum); //以每个结点做为根节点进行dfs
FindPath(root->left,sum); //递归
FindPath(root->right,sum);
return ans;
}
};