Given a binary tree and a sum, find all root-to-leaf paths where each path’s sum equals the given sum.
Note: A leaf is a node with no children.
Example:
Given the below binary tree and sum = 22,
5
/ \
4 8
/ /
11 13 4
/ \ /
7 2 5 1
Return:
[
[5,4,11,2],
[5,8,4,5]
]
大意:在112的基础上,增加了要输出符合条件的路径。
方法:两种方法
1.递归。和112题相似,还是需要用深度优先搜索DFS,只不过需要用到二维的vector,而且每当DFS搜索到新节点时,都要保存该节点(sum)。而且每当找出一条路径之后,都将这个保存为一维vector(res1)的路径保存到最终结果二位vector中(res)。并且,每当DFS搜索到子节点,发现不是路径和时,返回上一个结点时(sum-sum1),需要把该节点从一维vector中移除。
class Solution {
public:
vector<vector<int>> pathSum(TreeNode* root, int sum) {
vector<vector<int>> res;
vector<int> path;
helper(root,sum,res,path);
return res;
}
private:
void helper(TreeNode* node,int sum,vector<vector<int>> &res,vector<int> &path){
if(!node) return;
path.push_back(node->val);
if(sum==node->val && !node->left && !node->right){
res.push_back(path);
}
helper(node->left,sum-node->val,res,path);
helper(node->right,sum-node->val,res,path);
path.pop_back();
}
};