LeetCode热题100 路径总和 III

题目描述

给定一个二叉树的根节点 root ,和一个整数 targetSum ,求该二叉树里节点值之和等于 targetSum 的 路径 的数目。

路径 不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的(只能从父节点到子节点)。

示例 1:


输入 :root = 10,5,-3,3,2,null,11,3,-2,null,1, targetSum = 8
输出 :3
解释 :和等于 8 的路径有 3 条,如图所示。

示例 2:

输入 :root = 5,4,8,11,null,13,4,7,2,null,null,5,1, targetSum = 22
输出 :3

提示:

二叉树的节点个数的范围是 0,1000
−109<=Node.val<=109-10^9 <= Node.val <= 10^9−109<=Node.val<=109

-1000 <= targetSum <= 1000

思路

见代码。

代码

cpp 复制代码
/**
 * 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 {
public:
    unordered_map<long long, int>mp;
    int pathSum(TreeNode* root, int targetSum) {
        mp[0] = 1;
        return dfs(root, 0, targetSum);
    }

    int dfs(TreeNode *root, long long curr, int targetSum)
    {
        if(root == nullptr) return 0;
        // 当前结点的答案
        int res = 0;
        // 现在总和
        curr += root->val;
        // 若有和前面结点相加为targetSum,更新答案
        if(mp.count(curr - targetSum))
        {
            res = mp[curr - targetSum];
        }
        ++mp[curr];
        res += dfs(root->left, curr, targetSum);
        res += dfs(root->right, curr, targetSum);
        // 只在自己的子树内生效, 不允许在其他字数出现
        --mp[curr];
        return res;
    }
};
相关推荐
Pniubi1 小时前
关于算法用现成数据结构和手撕
数据结构
唐墨1233 小时前
linux kernel 比较数据大小的宏定义
linux·算法
拂拉氏3 小时前
【知识讲解-题目讲解】 统计满足K约束的子字符串数量2 (滑动窗口+前缀和+二分查找+分析)
算法·前缀和·二分查找·滑动窗口
通信小呆呆8 小时前
电子战抗干扰技术:当“不被干扰”成为一种系统
算法·通信·mimo·isac
老歌老听老掉牙8 小时前
案例 7:二维 Rosenbrock 函数最小值(香蕉函数)
c++·pso
笑容掩饰我的失落8 小时前
区块链从小白到精通(八):从零实现一个标准 ERC-20 代币(完整可运行代码)
算法·区块链·php·智能合约·哈希算法
Severus_black8 小时前
【C++初阶】Stack、Queue算法题大练兵!
c语言·c++·算法·stl
2601_9679791211 小时前
机器学习必修课:经典算法与Python实战
python·算法·机器学习·实战·课程
程序猿打工仔11 小时前
计算机基础(C语言)考编、专技岗
c语言·开发语言·算法
通信小呆呆11 小时前
电子战干扰技术:从“蛮力压制”到“精准欺骗”,再到“认知博弈”
算法·通信·isac