输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径
问题难点在于, 为了确定某一条路径是否符合要求,需要走完整条路来确定是否要保留之前走的路径点
我们采取递归寻找的方法,将走过的路径全部保留,如果最后发现这条路并不是我们需要的,则将已经存入的路径点弹出,如果满足,则将整条路径返回
代码如下
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
vector<vector<int> > res; //定义全局变量,方便结果传递
vector<int> temp;
void help(TreeNode* root,int expectNumber){
temp.push_back(root->val); //无论如何都将走过的节点值压入
if(root->left==NULL && root->right==NULL && expectNumber-root->val==0) res.push_back(temp); //如果满足条件,则返回这条路径
else{
if(root->left) help(root->left,expectNumber-root->val); //否则继续递归
if(root->right) help(root->right,expectNumber-root->val);
}
temp.pop_back(); //如果递归结束,res并没有返回,则从后开始弹出当前节点,重新走另一条路
}
public:
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
if(root) help(root,expectNumber);
return res;
}
};