题目描述
输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
思路:递归
全局变量保存要输出的路径数组和当前路径
要得到路径中结点和为给定值的目标路径,即在根结点的角度,得到左子树和右子树的目标路径(只不过它们的要求的数值和为 原本的目标值 减去 根结点自己的值)
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
private:
vector<vector<int> > res;
vector<int> path;
public:
void find (TreeNode* root,int expectNumber){
if(root == NULL) return ;
path.push_back(root->val);
if(!root->left && !root->right && root->val == expectNumber){//找到辣
//别直接用res.push_back(path);因为题目要求从大到小排序,所以用insert
for(int i = 0;i < res.size();i++){
if(path.size() > res[i].size())
res.insert(res.begin()+i,path);
}
if(res.empty() || path.size() <= res.back().size())
res.push_back(path);
}else{//还没到底,那就递归寻找
if(root->left)
find(root->left, expectNumber - root->val);
if(root->right)
find(root->right, expectNumber - root->val);
}
path.pop_back();//这时有俩结果:找和没找到,不管咋样都要把path的最后一个结点踢掉
}
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
find(root,expectNumber);
return res;
}
};