输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
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>> pathVectors;
vector<int> pathVector;
FindPath(root, expectNumber, pathVectors, pathVector);
return pathVectors;
}
void FindPath(TreeNode* root, int expectNumber, vector<vector<int>>& pathVectors, vector<int>&pathVector) {
do
{
if (root == NULL) {
break;
}
pathVector.push_back(root->val);
int sum = this->sum(pathVector);
if (sum == expectNumber && root->left == NULL && root->right == NULL) {
// 叶子节点, 且值为目标值
pathVectors.push_back(pathVector);
}
FindPath(root->left, expectNumber, pathVectors, pathVector);
FindPath(root->right, expectNumber, pathVectors, pathVector);
// 删除最后一个
pathVector.pop_back();
} while (0);
return;
}
int sum(vector<int>& v) {
int s = 0;
for (auto it : v) {
s += it;
}
return s;
}
};
测试