题目:
输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
思路:
使用递归方法,从根节点出发,逐步向子节点遍历,使用动态数组保存所经过的节点,并保存每一步时当前的路径节点之和,当到达叶节点时,该路径上所有节点之和如果等于目标值,则保存该路径,并返回父节点,去判断其他路径
代码如下:
/*
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>> result;
if(root == nullptr)
return result;
int cur_sum = 0;
vector<int> path;
FindPath(root, expectNumber, cur_sum, result, path);
return result;
}
void FindPath(TreeNode* root, int expectNumber, int cur_sum,
vector<vector<int>> &result, vector<int> &path)
{
cur_sum += root->val;
path.push_back(root->val);
if(cur_sum == expectNumber && root->left == nullptr
&& root->right == nullptr)
{
result.push_back(path); // 因为之后会逐步pop回去,所以这里不需要path.clear()
}
if(root->left != nullptr)
FindPath(root->left, expectNumber, cur_sum, result, path);
if(root->right != nullptr)
FindPath(root->right, expectNumber, cur_sum, result, path);
path.pop_back();
}
};