1.题目:
输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
2.解析:
要求输出所有的路径和情况,所以返回的结果是一个vector<vector<int>>
,既然是从根节点打印到叶子节点,相当于先序遍历每个节点,(结束的条件就是路径和为所需的,且该节点没有孩子节点)
注意,当最后的叶子节点不符合的时候一定要pop_back
掉
Code
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
private:
std::vector<int> tmp;
std::vector<vector<int>> res;
public:
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
if(root == NULL) return res;
tmp.push_back(root->val);
if((expectNumber-root->val == 0) && root->left == NULL && root->right == NULL)
res.push_back(tmp);
FindPath(root->left,expectNumber-root->val);
FindPath(root->right,expectNumber-root->val);
tmp.pop_back();
return res;
}
};