题目链接
题意
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
解题思路
由于从根节点,所以使用前序遍历。
记忆化搜索,注意其中对vector的操作。
若是要在遍历的同时打印满足条件的路径,由于STL中stack只能取到栈顶元素,所以应采用vector。
做了剪枝。
待做
暂时没有做括号中要求。
代码
/*
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> > FindPath(TreeNode* root,int expectNumber) {
if(root){
dfs(root,expectNumber);
}
return expectPath;
}
private:
vector<vector<int>> expectPath;
vector<int> path;
void dfs(TreeNode* root,int expectNumber){
path.push_back(root->val);
if(expectNumber-root->val==0&&!root->left&&!root->right){
expectPath.push_back(path);
}
if(expectNumber-root->val>0){
if(root->left){
dfs(root->left,expectNumber-root->val);
}
if(root->right){
dfs(root->right,expectNumber-root->val);
}
}
path.pop_back();
}
};