题目描述:
给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22
,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
[
[5,4,11,2],
[5,8,4,5]
]
代码:
class Solution {
public:
vector<vector<int>> pathSum(TreeNode* root, int sum) {
vector<vector<int>>ans;
vector<int>res;
if(root==NULL)return ans;
helper(root,res,ans,sum);
return ans;
}
void helper(TreeNode * root,vector<int>as,vector<vector<int> >&ans,int sum){
if(!root->left&&!root->right){
vector<int>res;
int s=0;
for(auto a:as){
res.push_back(a);
s+=a;
}
res.push_back(root->val);
if(s+root->val==sum)
ans.push_back(res);
}
if(root->left){
vector<int>res;
for(auto a:as)
res.push_back(a);
res.push_back(root->val);
helper(root->left,res,ans,sum);
}
if(root->right){
vector<int>res;
for(auto a:as)
res.push_back(a);
res.push_back(root->val);
helper(root->right,res,ans,sum);
}
}
};
赶时间,有些情况重复,所以代码看起来有点复杂。明天再改。