题目描述
输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
链接:https://www.nowcoder.com/practice/b736e784e3e34731af99065031301bca?tpId=13&tqId=11177&tPage=2&rp=2&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking
题解:很直白的递归求解,递归遍历各个节点,插入容器内,当满足题目要求(节点值的和为该输入的数据,并且最后一个结点为叶节点),将整个容器插入到另一个容器内保存好,返回上一个结点继续遍历,查看是否有其他情况。
AC代码:
/*
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> >res;
vector<int>v;
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
if(root)look_for(root,expectNumber); //当根节点不为空时,开始遍历,若为空,直接返回
return res;
}
void look_for(TreeNode* root,int expectNumber){
v.push_back(root->val); //每循环一次都将新值插入。
if(expectNumber-root->val==0&&root->left==NULL&&root->right==NULL) //符合题目条件(叶节点,路径上所有结点和等于输入的值)的插入到结果容器内。
res.push_back(v);
if(expectNumber>0){ //仅当输入的值大于零时,做递归
if(root->left)look_for(root->left,expectNumber-root->val);
if(root->right)look_for(root->right,expectNumber-root->val);
}
v.pop_back(); //弹出最顶上的那个结点值
}
};
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> >res;
vector<int>v;
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
if(root)look_for(root,expectNumber); //当根节点不为空时,开始遍历,若为空,直接返回
return res;
}
void look_for(TreeNode* root,int expectNumber){
v.push_back(root->val); //每循环一次都将新值插入。
if(expectNumber-root->val==0&&root->left==NULL&&root->right==NULL) //符合题目条件(叶节点,路径上所有结点和等于输入的值)的插入到结果容器内。
res.push_back(v);
if(expectNumber>0){ //仅当输入的值大于零时,做递归
if(root->left)look_for(root->left,expectNumber-root->val);
if(root->right)look_for(root->right,expectNumber-root->val);
}
v.pop_back(); //弹出最顶上的那个结点值
}
};