题目:
输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。路径定义为从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
1.解题思路:
- 用前序遍历的方式访问到某一个节点时,我们把该节点添加到路径上,并累加该节点的值;
- 如果该节点为叶节点,并且路径中节点值的和刚好等于输入的整数,则当前路径符合要求;
- 如果当前节点不是叶节点,则继续访问它的子节点;
- 当前节点访问结束时,递归函数将自动回到它的父节点,因此在函数退出之前要在路径上删除当前节点并减去当前节点的值。
2.代码:
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution{
private:
vector<vector<int>> result;//最终结果
vector<int> temp;//临时结果
public:
vector<vector<int>> FindPath(TreeNode* root, int expectNumber){
//如果二叉树为空
if(root == nullptr) return result;
//根节点赋值
temp.push_back(root->val);
//如果当前节点为叶子节点,且相加和刚好等于输入的整数,则该条路径满足条件
if(expectNumber == root->val && root->left == nullptr && root->right == nullptr){
result.push_back(temp);
}
//如果不是叶子节点,则遍历它的子节点
FindPath(root->left, expectNumber - root->val);
FindPath(root->right, expectNumber - root->val);
//返回父节点之前,在路径上删除当前节点
if(!temp.empty()) temp.pop_back();
return result;
}
};