1.题目
给你二叉树的根节点 root 和一个整数目标和 targetSum ,找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。
叶子节点 是指没有子节点的节点。
示例 1:
输入:root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出:[[5,4,11,2],[5,8,4,5]]
示例 2:
输入:root = [1,2,3], targetSum = 5
输出:[]
示例 3:
输入:root = [1,2], targetSum = 0
输出:[]
2.链接
题目:https://leetcode.cn/problems/path-sum-ii/
讲解:https://programmercarl.com/0112.%E8%B7%AF%E5%BE%84%E6%80%BB%E5%92%8C.html#%E6%80%9D%E8%B7%AF
3.解题思路
参考二叉树路径总和第1题。本题要遍历整个树,找到所有路径,所以递归函数不要返回值!
如图:
这道题可以使用深度优先搜索(DFS)来解决。我们可以从根节点开始遍历,每遍历到一个节点,就将它加入到路径中,同时将目标值减去该节点的值。当遍历到叶子节点时,如果目标值恰好为0,则说明我们找到了一条符合题意的路径,将该路径加入到结果中。如果目标值不为0,则说明该路径不符合题意,直接返回。在遍历完当前节点的左右子树后,需要将该节点从路径中删除,同时将目标值加上该节点的值,回溯到上一层节点。具体实现可以看代码注释。
4.代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
//递归法
class Solution {
private:
vector<vector<int>> result;
vector<int> path;
//递归函数不需要返回值,因为我们要遍历整个树
void traversal(TreeNode* cur, int count) {
// 遇到了叶子节点且找到了和为sum的路径
if (!cur->left && !cur->right && count == 0) {
result.push_back(path);
return ;
}
// 遇到叶子节点而没有找到合适的边,直接返回
if (!cur->left && !cur->right) return ;
if (cur->left) { // 左 (空节点不遍历)
path.push_back(cur->left->val);//将左节点的值加入path动态数组
count -= cur->left->val;//计数器减左节点值,目的找到符合题意的路径
traversal(cur->left, count);//递归
count += cur->left->val;//计数值回溯
path.pop_back();//动态数组储存的路径点信息回溯
}
if (cur->right) {
path.push_back(cur->right->val);
count -= cur->right->val;
traversal(cur->right, count);
count += cur->right->val;
path.pop_back();
}
return ;
}
public:
vector<vector<int>> pathSum(TreeNode* root, int targetSum) {
result.clear();
path.clear();
if (root == nullptr) return result;
path.push_back(root->val);//把根节点放进路径
traversal(root, targetSum - root->val);//递归
return result;
}
};