【LeetCode热题100(47/100)】路径总和 III

题目地址: 链接

思路: 通过前缀和 + 回溯,每次记录当前前缀和,每次记录当前离 targetSum 的偏移量, 统计当前满足条件的路径数量。

js 复制代码
/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @param {number} targetSum
 * @return {number}
 */
var pathSum = function(root, targetSum) {
    let map = new Map();
    let ans = 0;
    function dfs(root, pre) {
        if(!root) return null;
        
        let ppre = pre + root.val;
        let check_num = ppre - targetSum; // 偏移量

        ans += map.get(check_num) || 0;
        map.set(ppre, map.get(ppre) + 1 || 1);

        dfs(root.left, ppre);
        dfs(root.right,ppre);
        map.set(ppre, map.get(ppre) - 1);
    }
    map.set(0, 1);
    dfs(root, 0);
    return ans;
};
相关推荐
smj2302_796826524 小时前
解决leetcode第3721题最长平衡子数组II
python·算法·leetcode
m0_626535204 小时前
力扣题目练习 换水问题
python·算法·leetcode
第六五4 小时前
DPC和DPC-KNN算法
人工智能·算法·机器学习
一匹电信狗4 小时前
【LeetCode_160】相交链表
c语言·开发语言·数据结构·c++·算法·leetcode·stl
再卷也是菜5 小时前
C++篇(14)二叉树进阶算法题
c++·算法
小邓儿◑.◑5 小时前
贪心算法 | 每周8题(三)
算法·贪心算法
2401_841495645 小时前
【数据结构】最长的最短路径的求解
java·数据结构·c++·python·算法·最短路径·图搜索
小龙报5 小时前
《算法每日一题(1)--- 连续因子》
c语言·开发语言·c++·windows·git·算法·visual studio
夜晚中的人海6 小时前
【C++】滑动窗口算法习题
开发语言·c++·算法