问题描述:
输入一棵二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
样例:
找出所有符合条件的路径,需要建立两个vector,一个一维的path(存一条满足条件的路径),一个二维的ans(存满足所有条件的路径),如果到最后sum的值为0,那么就要将该路径path加入到ans中,否则继续遍历,如果ans将path加入到路径中后,需要将path清空。如果sum不满足条件的话,那么path会被清空。
class Solution {
public:
vector<vector<int>> ans;
vector<int> path;
vector<vector<int>> findPath(TreeNode* root, int sum) {
bfs(root,sum);
return ans;
}
void bfs(TreeNode* root,int sum)
{
if(!root) return;
sum-=root->val;
path.push_back(root->val);
if(!root->left&&!sum&&!root->right) ans.push_back(path);
bfs(root->left,sum);
bfs(root->right,sum);
path.pop_back();
}
};