每日一道Leetcode - 560. 和为K的子数组【前缀和+字典】

奋进的杰西

于 2021-08-05 12:00:47 发布

阅读量373

点赞数

分类专栏： Leetcode

本文链接：https://blog.youkuaiyun.com/weixin_41041275/article/details/119414931

版权

Leetcode 专栏收录该内容

304 篇文章

订阅专栏

这个博客讨论了一种算法，用于计算一个整数数组中和为k的连续子数组的数量。通过维护前缀和字典，可以有效地查找目标和的子数组。这种方法在O(n)的时间复杂度内完成，对于给定的输入数组nums和目标值k，返回了满足条件的子数组个数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述

class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
        # 前缀和字典，当前前缀和:次数; 前缀和为0的数组出现次数为1次
        preSum = {}
        preSum[0] = 1
        count = 0
        pre = 0
        for num in nums:
            pre += num
            if pre-k in preSum:
                count += preSum[pre-k]
            if pre not in preSum:
                preSum[pre] = 0
            preSum[pre] += 1
        return count