输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
需要注意,这道题的题意,路径必须是根节点到某叶子节点。根节点到中间节点,即使路径和满足也不行
此处做下标记,代码注释我已经写出如果要更完善点,应该加分支限定,此文代码需要更新
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
#include <vector>
class Solution {
public:
vector<vector<int> > Res;
vector<int> path;
void UpdataRes(TreeNode* root,int expectNumber)
{
if(root==nullptr)
return;
//path.push_back(root->val);
//expectNumber-=root->val;
//if(expectNumber>=root->val) //这样写有问题,若当前的expectNumber小于当前节点的val,会越过当前节点去访问当前节点的子节点
//题意希望这种情况应该终止改路径并回溯
//{
// path.push_back(root->val);
// expectNumber-=root->val;
//}
path.push_back(root->val); //不管root->val多大直接放入,如果大了,expectNumber会一直是负(我觉得要再好点
//应该加分支限定当expectNumber为负数时候,就终止该条路径并回溯。该条路径不会进Res
expectNumber-=root->val;
if(!root->left&&!root->right&&!expectNumber)
{
Res.push_back(path);
}
else
{
if(root->left!=nullptr&&expectNumber)
{
FindPath(root->left,expectNumber);
}
if(root->right!=nullptr&&expectNumber)
{
FindPath(root->right,expectNumber);
}
}
path.pop_back();
}
vector<vector<int> > FindPath(TreeNode* root,int expectNumber)
{
UpdataRes(root,expectNumber);
return Res;
}
};