题目描述
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
算法设计
用深度搜索,逐个遍历二叉树的结点。要注意的是题目中说,路径是从树的根结点开始往下一直到叶节点,因此如果还未到叶节点,值的和就与expectNumber相等的话是不能算做一个正确结果的。
C++代码
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
void func(TreeNode *root, int expectNumber, set<vector<int> > &res_set, vector<int> nums){
vector<int> res = nums;
if(root == nullptr){
return;
}
int exp = expectNumber - root->val;
res.push_back(root->val);
if(exp < 0){
return;
}
else if(exp == 0){
if(root->left == nullptr && root->right == nullptr)
res_set.insert(res);
return;
}
else{
func(root->left, exp, res_set, res);
func(root->right, exp, res_set, res);
}
}
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
set<vector<int> > res_set;
vector<vector<int> > res;
vector<int> nums;
func(root, expectNumber, res_set, nums);
set<vector<int> >::iterator it = res_set.begin();
for(it;it!=res_set.end();it++){
res.push_back(*it);
}
return res;
}
};