版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/kirito0104/article/details/82108953
给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22
,
5 / \ 4 8 / / \ 11 13 4 / \ \ 7 2 1
返回 true
, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2
。
思路
1、采取DFS
2、用vector<int>形参来递归,用vector<vector<int>>引用形参来达到随时push_back的目的
3、内存要求大,但是应该可以
code
class Solution {
public:
vector<vector<int>> pathSum(TreeNode* root, int sum) {
vector<vector<int>> vv;
if(!root)
return vv;
vector<int> v;
path(root, v, sum, vv);
return vv;
}
void path(TreeNode * root, vector<int> v, int a, vector<vector<int>> &vv){
int sum = accumulate(v.begin(),v.end(),0);
if(!root && sum==a){
vv.push_back(v);
return;
}
if(!root && sum != a)
return;
v.push_back(root->val);
if(root->left && root->right){
path(root->left, v, a, vv);
path(root->right, v, a, vv);
}
else if(root->left && !root->right)
path(root->left, v, a, vv);
else
path(root->right, v, a, vv);
}
};