题目描述
输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
很明显改用回溯法:
/*
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> > Paths;
vector<int> path;
void sum_path(TreeNode* root, int num)
{
//终止条件
if (root->left == NULL && root->right == NULL)
{
if (root->val == num)
{
path.push_back(root->val);
Paths.push_back(path);
path.pop_back();
}
return;
}
//回溯法
//递归主体
path.push_back(root->val);
if (root->left !=NULL)
sum_path(root->left, num - root->val);
if (root->right !=NULL)
sum_path(root->right,num - root->val);
//回溯法
path.pop_back();
}
public:
vector<vector<int> > FindPath(TreeNode* root, int expectNumber) {
if(root)/*记得判断边界*/
sum_path(root, expectNumber);
return Paths;
}
};