二叉树中和为某一值的路径

时间限制：1秒空间限制：32768K
通过比例：24.42%
最佳记录：0ms|8552K（来自 牛客688826号）

题目描述

输入一颗二叉树和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。

/*
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) {
vector<vector<int> > pathList;
if(!root) return pathList;
if(root->val>expectNumber) return pathList;
vector<int> sublist;
int curNum=0;
NLRSearch(root,pathList,sublist,curNum,expectNumber);
return pathList;
}
void NLRSearch(TreeNode* root,vector<vector<int> >& pathList,vector<int>& sublist,int curNum,const int& expectNumber){
if(!root) return;
curNum+=root->val;
sublist.push_back(root->val);
if(expectNumber==curNum){
if(!root->left&&!root->right){
pathList.push_back(sublist);
}
}else{
NLRSearch(root->left,pathList,sublist,curNum,expectNumber);
NLRSearch(root->right,pathList,sublist,curNum,expectNumber);
}
sublist.pop_back();
}
};