题目来源:牛客网-剑指offer
题目描述:
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
自己总结的github地址:
https://github.com/Rayholle/LeetCode-Newcoder/blob/master/Tree/判断树路径上的和是否为某个数.cpp
思路分析:路径问题从树的根结点到叶结点,所以需要深度优先遍历(dfs),在递归过程中发现当前路径和为给定的数时,将该路径存入vector中,再遍历其左子树与右子树。
细节:
1、dfs函数参数:
需要树结点,二维数组(最终返回值),一维数组(路径trace),当前需要值。
所以我们声明dfs函数为:注意res和trace用引用传递
void dfs(TreeNode* root, vector<vector<int>>& res, vector<int>& trace, int s);
2、关于dfs的函数定义:
需要先将该结点的val存储在trace,在函数定义结尾,需要将该值从trace中pop出去(当前递归层结束)。
void dfs(TreeNode* root, vector<vector<int>>& res, vector<int>& trace, int s){
trace.push_back(root->val);
if(!root->left&&!root->right){//当root为叶结点时,题目要求路径为从根到叶结点
if(s==root->val)
res.push_back(trace);
if(root->left)
dfs(root->left, res, trace, s-root->val);
if(root->right)
dfs(root->right, res, trace, s-root->val);
trace.pop_back(root->val);
}
接下来是完整代码:
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
vector<int> trace;
vector<vector<int>> ret;
if(root)
dfs(root, expectNumber, trace, ret);
return ret;
}
void dfs(TreeNode* root,int S, vector<int>& trace, vector<vector<int>>& ret){
trace.push_back(root->val);
//判断是否为叶子节点
if(!root->left && !root->right){
if(S == root->val)
ret.push_back(trace);
}
if(root->left)
dfs(root->left, S-root->val, trace, ret);
if(root->right)
dfs(root->right, S-root->val, trace, ret);
trace.pop_back();
}