深度优先遍历题目的变种。只需要在遍历时多传两个参数,一个用于记录路径和,一个用于记录遍历过的结点即可。
这里有个小坑需要注意,最开始在递归传参时使用了引用传参,导致递归返回后参数不回还原,需要改为值传参。果然引用传参还是存在危险的,需要稍用。
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
vector<vector<int>> result;
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
if(root){
vector<int> arr;
go(root, 0, arr, expectNumber);
}
return result;
}
void go(TreeNode* p, int sum, vector<int> arr, int ex){
int ns = sum + p->val;
vector<int> nv = arr;
nv.push_back(p->val);
if(!p->left && !p->right){
if(ns == ex){
result.push_back(nv);
}
}
if(p->left){
go(p->left, ns, nv, ex);
}
if(p->right){
go(p->right, ns, nv, ex);
}
}
};