【题目】
输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
【思路】
首先,理解题意,要我们找的是从根节点开始往下到叶节点所经过的节点形成一条路径,满足两点:
(1)这条路径要有根节点和一个叶节点。
(2)路径的节点值之和要等于题目要求的输入整数。
所以思路很简单,从根节点开始往下遍历,使用深度优先搜索 。
【代码】
- 写法一
先设置tmp和result两个全局变量,tmp用于存放临时结果,result用于存放最终结果。
每次更新expectNumber的值(减去经过的节点的值),最后一次相减结果为0说明找到一个结果。
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(nullptr), right(nullptr) {
}
};*/
class Solution {
public:
vector<vector<int> > FindPath(TreeNode* root, int expectNumber) {
if (root == nullptr) {
return result;
}
tmp.push_back(root->val);
// 路径和为expectNumber,并且为叶子节点
if ((expectNumber - root->val) == 0 && root->left == nullptr && root->right == nullptr) {
result.push_back(tmp);
}
// 遍历子节点
FindPath(root->left, expectNumber - root->val);
FindPath(root->right, expectNumber - root->val);
tmp.pop_back(); // 返回父节点前先删除当前节点
return result;
}
private:
vector<int> tmp;
vector<vector<int>> result;
};
- 写法二
没有回溯。多了一个currentSum > expectNumber剪枝判断,减少搜索量。
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(nullptr), right(nullptr) {
}
};*/
class Solution {
public:
vector<vector<int> > FindPath(TreeNode* root, int expectNumber) {
vector<int> tmp;
GetPath(root, expectNumber, tmp, 0);
return result;
}
void GetPath(TreeNode* root, int expectNumber, vector<int> tmp, int currentSum){
if (currentSum > expectNumber || root == nullptr) {
return;
}
currentSum += root->val;
tmp.push_back(root->val);
// 路径和为expectNumber,并且为叶子节点
if (expectNumber == currentSum && root->left == nullptr && root->right == nullptr) {
result.push_back(tmp);
return;
}
// 遍历子节点
GetPath(root->left, expectNumber, tmp, currentSum);
GetPath(root->right, expectNumber, tmp, currentSum);
}
private:
vector<vector<int>> result;
};