题目描述:
给定一棵二叉树,其中每个节点都含有一个整数数值(该值或正或负)。设计一个算法,打印节点数值总和等于某个给定值的所有路径的数量。注意,路径不一定非得从二叉树的根节点或叶节点开始或结束,但是其方向必须向下(只能从父节点指向子节点方向)。
示例:
思路及题解:
本题与面试题04.10很像,具体参见我的另一篇博客:检查子树
不同的是:本题需要遍历每一个节点以及对每一个节点进行dfs,所以时间复杂度为O(n^2),空间复杂度为O(1)。
代码及运行评价:
/**
* 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 {
int res = 0;
public:
int pathSum(TreeNode* root, int sum) {
if(root==NULL)return 0;
helper(root,sum,0);
pathSum(root->left,sum);
pathSum(root->right,sum);
return res;
}
void helper(TreeNode* node,int sum,int temp)
{
if(node==NULL)return;
temp+= node->val;
if(sum==temp)
{
res++;
}
helper(node->left,sum,temp);
helper(node->right,sum,temp);
}
};