二叉树路径总和第2题

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;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值