python-leetcode-路径总和 III

437. 路径总和 III - 力扣(LeetCode)

python 复制代码
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def pathSum(self, root: Optional[TreeNode], targetSum: int) -> int:
        prefix_sum = defaultdict(int)  # 记录前缀和出现的次数
        prefix_sum[0] = 1  # 初始状态,前缀和为0出现一次

        def dfs(node, curr_sum):
            if not node:
                return 0
            
            curr_sum += node.val  # 更新当前前缀和
            count = prefix_sum[curr_sum - targetSum]  # 计算符合条件的路径数
            
            prefix_sum[curr_sum] += 1  # 记录当前前缀和
            count += dfs(node.left, curr_sum)  # 递归左子树
            count += dfs(node.right, curr_sum)  # 递归右子树
            
            prefix_sum[curr_sum] -= 1  # 回溯,撤销当前节点的前缀和
            
            return count
        
        return dfs(root, 0)
相关推荐
HXhlx14 小时前
CART决策树基本原理
算法·机器学习
Wect14 小时前
LeetCode 210. 课程表 II 题解:Kahn算法+DFS 双解法精讲
前端·算法·typescript
颜酱15 小时前
单调队列:滑动窗口极值问题的最优解(通用模板版)
javascript·后端·算法
Gorway1 天前
解析残差网络 (ResNet)
算法
拖拉斯旋风1 天前
LeetCode 经典算法题解析:优先队列与广度优先搜索的巧妙应用
算法
Wect1 天前
LeetCode 207. 课程表:两种解法(BFS+DFS)详细解析
前端·算法·typescript
灵感__idea1 天前
Hello 算法:众里寻她千“百度”
前端·javascript·算法
Wect2 天前
LeetCode 130. 被围绕的区域:两种解法详解(BFS/DFS)
前端·算法·typescript
NAGNIP2 天前
一文搞懂深度学习中的通用逼近定理!
人工智能·算法·面试
颜酱3 天前
单调栈:从模板到实战
javascript·后端·算法