题目描述:输入一棵二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
分析:
完整代码:
/*
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) {
//声明一个二维数组
vector<vector<int> > path;
//声明一个临时栈
vector<int>tmp;
//检查输入的合法性
if(root == nullptr)
return path;
//声明一个变量用来表示路径节点值的和
int current =0;
//调用路径寻找函数
findpath(root, expectNumber, current, path,tmp);
//返回最终的路径
return path;
}
private:
void findpath(TreeNode* root, int expectNumber, int current,vector<vector<int> >&path,vector<int>&tmp)
{
//计算当前路径节点值的和
current += root->val;
//将访问过节点的值压入栈中
tmp.push_back(root->val);
//判断当前节点是否为叶节点
bool IsLeaf = root->left == nullptr && root->right == nullptr;
/******判断当前节点是否为叶节点,并且路径节点值得和是否等于期望值******/
if(IsLeaf && current == expectNumber)
{
//将某一路径的栈,压入到二维的栈中
path.push_back(tmp);
}
//判断当前节点的左子树是否为空
if(root->left != nullptr)
findpath(root->left, expectNumber, current, path,tmp);
//判断当前节点的右子树是否为空
if(root->right != nullptr)
findpath(root->right, expectNumber, current, path,tmp);
//删除当前的节点
tmp.pop_back();
//更新路径节点值的和
current -= root->val;
}
};