给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
[
[5,4,11,2],
[5,8,4,5]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/path-sum-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
/**
* 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:
vector<vector<int>>all;
vector<vector<int>> pathSum(TreeNode* root, int sum) {
if(root==NULL)
return all;
vector<int> he;
hasPathSumHelper(root,sum,0,he);
return all;
}
void hasPathSumHelper(TreeNode* root,int sum,int num,vector<int> he)
{
if(root->val+num==sum&&root->left==NULL&&root->right==NULL)//题目要求是根节点到叶子节点,并不是任一节点
{
he.push_back(root->val);
all.push_back(he);
return ;
}
if(root->left!=NULL)
{
he.push_back(root->val);
hasPathSumHelper(root->left,sum,num+root->val,he);
he.pop_back();//需要把刚刚加入得不对得节点值取出来,不需要改变num是因为num得值并没有变,传入得值是num加上另一个数
}
//he.pop_back();//但是放在这里就不行,因为上面那个if不一定会执行,这样子弹出得数据就是上一个得了
//尤其是当数组里还没有值得时候,会出现下表越界错误
if(root->right!=NULL)
{
he.push_back(root->val);
hasPathSumHelper(root->right,sum,num+root->val,he);
}
return ;
}
};