题目描述:
给定一个二叉树,它的每个结点都存放着一个整数值。
找出路径和等于给定数值的路径总数。
路径不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的(只能从父节点到子节点)。
二叉树不超过1000个节点,且节点数值范围是 [-1000000,1000000] 的整数。
示例:
解题思路:
首先我们想一下,如果是求 “从根节点出发,求根节点到任意节点和为sum的路径总数”,这个问题的解法很简单,我们进行递归求解即可,那么这里可以从任意节点出发,我们就可以把每个节点看做起点,对其进行递归求解,但这对每个节点都要进行两次递归。
两次递归代码实现如下:
class Solution {
public:
int pathSum(TreeNode* root, int sum) {
if (root == nullptr)
return 0;
return pathSumCore(root, sum) +
pathSum(root->left, sum) +
pathSum(root->right, sum);
}
private:
int pathSumCore(TreeNode* root, int sum)
{
if (root == nullptr)
return 0;
sum -= root->val;
if (0 == sum) {
return 1 + pathSumCore(root->left, sum) +
pathSumCore(root->right, sum);
}
return pathSumCore(root->left, sum) +
pathSumCore(root->right, sum);
}
};
下面再讲一种一次递归求解的方法:
- 思想是遍历到某个节点时,检查该节点到根节点的路径上包含该节点的所有可能,如此每一个节点在检查可能的路径时都不会重复。
- 比如假设有路径1,2,3,4,-4,5, sum=5,如果遍历到3时,只需要检查1,2,3中是否有包含3的解,发现2,3是一个解;
- 遍历到5时只需检查包含5的解,包括 5,以及 4,-4,5 两个解。这样就没有重复检查。因为包含5的解在之前一定没被检查过。
- 检查方式也是极其简单,使用vector保存遍历过程中的值,然后反向for循环这个vector求和即可
一次递归代码实现:
class Solution {
public:
int pathSum(TreeNode* root, int sum) {
vector<int> path;
int res = 0;
pathSumCore(root, sum, path, res);
return res;
}
private:
void pathSumCore(TreeNode* root, int sum, vector<int>& path, int& res)
{
if (root == nullptr)
return;
path.push_back(root->val);
int curSum = 0;
for (int i = path.size()-1; i >= 0; --i)
{
curSum += path[i];
if (curSum == sum)
++res;
}
pathSumCore(root->left, sum, path, res);
pathSumCore(root->right, sum, path, res);
path.pop_back();
}
};