题目描述
输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
利用递归的思想,如果从根节点的路径和为n,那么就是找从根节点左子树路径和为n-root->val以及右子树路径和为n-root->val。并设置flag标识,一旦出现flag不唯一的,直接返回。不做任何事。
自认为比剑指offer写的好。逻辑清楚不易出错。
/*
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> > FindPath(TreeNode* root,int expectNumber) {
if(root==NULL)
return vector<vector<int> >{};
vector<vector<int> > res;
int flag=1;
recurFindPath(root,expectNumber,res,flag);
return res;
}
void recurFindPath(TreeNode* root,int expectNumber,vector<vector<int> > &vec,int flag) {
if(flag==0)
{return;}
if(root->left==NULL&&root->right==NULL&&expectNumber==root->val){
vec=vector<vector<int> >{vector<int>{root->val}};
return;
}
if(root->left==NULL&&root->right==NULL&&expectNumber!=root->val){
flag=0;return;
}
vector<vector<int>> left{};
vector<vector<int>> right{};
if(root->left)
recurFindPath(root->left,expectNumber-root->val,left,flag);
if(root->right)
recurFindPath(root->right,expectNumber-root->val,right,flag);
if(!left.empty()){
for(auto &c:left){
c.insert(c.begin(),root->val);
vec.push_back(c);
}
}
if(!right.empty()){
for(auto &c:right){
c.insert(c.begin(),root->val);
vec.push_back(c);
}
}
}
};