题目描述
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
思路:
利用前序遍历,将节点的值保存在容器中,便利到叶子节点时检查路径中节点的和是否满足条件,如果满足就加入最终的vector。然后回溯继续便利下一个叶子时将容器中的栈顶元素移除。
class Solution {
public:
vector<vector<int> > FindPath(TreeNode* root, int expectNumber) {
if (NULL == root)
return ret;
tmp.push_back(root->val);
if (expectNumber == root->val && root->left == NULL && root->right == NULL)//不便利到底,而是只遍历到叶子
ret.push_back(tmp);
FindPath(root->left, expectNumber - root->val);
FindPath(root->right, expectNumber - root->val);
tmp.pop_back();//在回溯到上一节点时pop最后一个元素。
return ret;
}
private:
vector<vector<int> > ret;
vector<int> tmp;
};