题目: Given a binary tree and a sum, find all root-to-leaf paths where each path’s sum equals the given sum.
题目的大意就是给你一棵二叉树,以及某个数sum,要你在这个二叉树上找到一条路径使得它所经过节点的值的和等sum。
解题思路: 思路比较清晰,是一课二叉树,求路径和,DFS!但是要记录路径,一开始想的是自己要弄个vector,经过一个点就把它push进去,到了叶子节点就判断一下有没有得到这样的结果,是就把它存起来,不是就pop返回上一层,但是这样的细节好多啊!还要标记那些访问过了,这样又可能要先遍历一遍。但是函数参数传递就直接提供了这样的一个记录路径的方法。这道题目还是比较简单的,具体看代码
vector<vector<int> > ans;
vector<vector<int>> pathSum(TreeNode* root, int sum)
{
vector<int> te;
path(root,sum,te);
return ans;
}
void path(TreeNode* root, int sum ,vector<int> p)
{
if(root==NULL)
return;
if(root->left==NULL&&root->right==NULL)
{
if(root->val==sum)
{
p.push_back(root->val);
ans.push_back(p);
return;
}
return;
}
else
{
p.push_back(root->val);
path(root->left,sum-root->val,p);
path(root->right,sum-root->val,p);
}
}