题目描述
输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
/*
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> >a;
vector<int> b;
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
if(root==NULL)
return a;
b.push_back(root->val);
if(root->left==NULL && root->right==NULL && (root->val-expectNumber)==0)
a.push_back(b);
FindPath(root->left,expectNumber-root->val);
FindPath(root->right,expectNumber-root->val);
if(b.size()!=0)
b.pop_back();
return a;
}
};
这道题想着不用递归,发现很麻烦,思路大体上是:再定义一个结构体,如下:
struct node
{
struct TreeNode *root;
int flag;
}
然后借助一个栈,每次访问一个节点就入栈,并把flag置1;这个标志是为了防止重复访问一个节点。看了别人写的递归,发现好简单。