1、题目描述
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
2、解题思路
可以使用一个栈来保存二叉树的路径,使用先序遍历的方式遍历二叉树。当遍历到一个节点时,将这个节点压入栈中。接着判断这个节点是否是叶子节点,如果是,计算栈中所有节点的值是否为目标整数,接着返回父节点,将叶子节点值从栈中弹出。如果不是叶子节点,当左子树为真时,则递归遍历左子树,如果右子树为真时,递归遍历右子树。递归代码如下:
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
vector<vector<int>> output;
if(root==nullptr) //由于没有考虑空指针的情况,很容易发生段错误
return output;
vector<int> path; //使用一个vector来保存路径
int cur_sum = 0; //用一个变量来保存当前路径和
find(root,expectNumber,cur_sum,path,output);
return output;
}
void find(TreeNode *p,int expectNumber, int &cur_sum,vector<int> &path,vector<vector<int>> &output){
cur_sum+=p->val;
path.push_back(p->val);
if(p->left==nullptr&&p->right==nullptr&&cur_sum==expectNumber){
output.push_back(path);
}
if(p->left)
find(p->left,expectNumber,cur_sum,path,output);
if(p->right)
find(p->right,expectNumber,cur_sum,path,output);
cur_sum-=p->val;
path.pop_back(); //返回父节点时路径中弹出当前节点值
return;
}