113. 路径总和 II
题目描述:
给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22,
思路:本菜鸡终于想到了dfs深搜,沿途保留所有结点路径,遇到符合条件的保存路径
代码:
class Solution {
vector<vector<int>> res;
vector<int> nums;
//int count=0;
public:
vector<vector<int>> pathSum(TreeNode* root, int sum) {
dfs(root, sum,0);
return res;
}
void dfs(TreeNode* root, int sum,int count)
{
if(!root)
return ;
count+=root->val;
nums.push_back(root->val);
if(count==sum&&(root->right==NULL&&root->left==NULL))
{
res.push_back(nums);
count=0;
// nums.clear();
}
dfs(root->left,sum,count);
dfs(root->right,sum,count);
nums.pop_back();
}
};
小结:判断条件可以用加法直至出现等于sum的时候,也可以直接用sum减,但是要注意用加法时记得传参,编写递归函数传参问题要多多注意。
在保存路径的过程中,无需清空nums,因为是在遍历过程中有一个符合题意得就存进答案数组一个,无需从头开始。