Leetcode | 437. 路径总和 III
给定一个二叉树,它的每个结点都存放着一个整数值。
找出路径和等于给定数值的路径总数。
路径不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的(只能从父节点到子节点)。
二叉树不超过1000个节点,且节点数值范围是 [-1000000,1000000] 的整数。
示例
题目解释
由题意可知,即从二叉树找随意找到一个由相连节点组成的路径的和等于sum值即可,找出所有可能的路径个数,返回总和。
解答
- 这种类型的题目,个人反应肯定第一解决方案选择深度优先搜索。
- 当当前节点的值刚好等于sum值,则返回1个可能值,如果不等,则更新新的sum值,即sum=sum-node->val
- 但是也要注意一点的就是,题目中说到不需要从根节点开始,也不需要在叶子节点结束,所以二叉树中的所有非空节点都可称为深度优先搜索的起始节点。
代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int pathSum(TreeNode* root, int sum) {
// 如果当前节点为空,则相当于未找到路径,返回0
if (!root)
return 0;
// 因为题目说,不要求一定从根节点开始和不一定从叶子节点结束
// 所以,二叉树中的所有节点都可以当作dfs的其实节点。
return dfs(root, sum) + pathSum(root->left, sum) + pathSum(root->right, sum);
}
int dfs(TreeNode* root, int sum){
// 如果当前节点为空,则相当于未找到路径,返回0
if (!root)
return 0;
// 如果当前节点的值刚好等于目标值,则返回加1
if (root->val == sum)
return 1 + dfs(root->left, 0) + dfs(root->right, 0);
// 如果当前节点的值不等于目标值,则更新新的目标值new_sum = sum - root->val;
return dfs(root->left, sum - root->val) + dfs(root->right, sum - root->val);
}
};