题目
给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22
,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
[
[5,4,11,2],
[5,8,4,5]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/path-sum-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析(DFS)
题目比较简单,用DFS即可。
- 比较简单的前序遍历二叉树,每层递归
sum
- 上一个节点的val
; - 遍历到叶子节点时判断sum的值是否等于0,等于0则记录当前路径,否则当前路径不符合要求,再遍历下个节点即可。
代码
class Solution {
public:
vector<vector<int>> result;
vector<vector<int>> pathSum(TreeNode* root, int sum) {
vector<int> tmp;
if (root == nullptr) {
return result;
}
dfs(root, sum, tmp);
return result;
}
void bfs(TreeNode *root, int sum, vector<int> &tmp)
{
if (root->left == nullptr && root->right == nullptr) {
sum -= root->val;
tmp.push_back(root->val);
if (sum == 0) {
result.push_back(tmp);
}
tmp.erase(--tmp.end());
return;
}
tmp.push_back(root->val);
if (root->left != nullptr) {
dfs(root->left, sum - root->val, tmp);
}
if (root->right != nullptr) {
dfs(root->right, sum - root->val, tmp);
}
tmp.erase(--tmp.end());
}
};