题目描述
输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
解题思路
由于路径是从根结点出发到叶结点,也就是说路径总是以根结点为起始点,因此我们首先需要遍历根结点,只有前序遍历是以根结点出发的,我们每访问到一个结点的时候,就把当前的结点添加到路径中去。而每一次当从子结点回到父结点的时候,我们都需要在路径上删除子结点。
为了确定某一条路径是否符合要求,需要走完整条路来确定是否要保留之前走的路径点。
我们采取递归寻找的方法,将走过的路径全部保留,如果最后发现这条路并不是我们需要的,则将已经存入的路径点弹出,如果满足,则将整条路径返回。
参考代码
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
//因为如果放在里面的话每次递归的时候就会重新new一个listALL和list,这样会把上一步的结果覆盖,所以必须写在外面
vector<vector<int>> resAll;
vector<int> path;
public:
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
if(root == NULL)
return resAll;
path.push_back(root->val);
expectNumber -= root->val;
if(expectNumber == 0 && root->left ==NULL && root->right == NULL)
resAll.push_back(path);
FindPath(root->left,expectNumber);
FindPath(root->right,expectNumber);
path.pop_back(); //移除最后一个元素,深度遍历完一条路径后要回退
return resAll;
}
};