题目描述
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
题目要求的路径是从根节点到叶子节点的路径。
即需要遍历整棵树,而且需要首先遍历根节点,所以采用前序遍历。
首先定义一个存储所有路径的vector<vector<int.> P,再定义一个存储符合要求的路径点的path,二叉树如下图所示,其中目标和为22,首先遍历根节点,再遍历左子树,10—5—4,path中压入了10—5—4,到达叶子节点,发现不符合条件,path删除4,去遍历5的右子树,10—5—7,path中压入的数字为10,5,7,此时符合和为22,所以将path压入P中。此时,左子树已经没有节点了,然后向上返回,首先删除7,再删除5。
此时,根节点的右子树不为空,则开始遍历右子树,10—12,path中压入了22,此时和满足22,所以将path压入P中,向上返回,path删除22,此时函数执行完毕,P中压入了两条路径:10—5—7与10—12。
代码如下:
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
void Paths(TreeNode* root,int curSum,int tarSum,vector<int> path)
{
path.emplace_back(root->val);
curSum+=root->val;
if(root->left==NULL&&root->right==NULL&&curSum==tarSum)//左右子树皆空且当前和等
//于目标和,则将当前路径压
{
vec.emplace_back(path);
}
if(root->left!=NULL) //左子树不为空,则进左子树中遍历查找
{
Paths(root->left,curSum,tarSum,path);
}
if(root->right!=NULL)
{
Paths(root->right,curSum,tarSum,path);
}
path.pop_back();
}
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
if(root==NULL)
{
return vec;
}
vector<int> path;
Paths(root,0,expectNumber,path);
return vec;
}
private:
vector<vector<int>> vec;
};