和为k的子数组（中等题）

最新推荐文章于 2024-08-10 22:52:22 发布

HUSTer-deeplearning

最新推荐文章于 2024-08-10 22:52:22 发布

阅读量52

点赞数

分类专栏：力扣刷题记录 python 文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/HUTAOhh/article/details/127006142

版权

力扣刷题记录同时被 2 个专栏收录

14 篇文章 0 订阅

订阅专栏

python

12 篇文章 0 订阅

订阅专栏

class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
        # 要求的连续子数组
        count = 0
        n = len(nums)
        preSums = collections.defaultdict(int)
        preSums[0] = 1

        presum = 0
        for i in range(n):
            presum += nums[i]
            
            # if preSums[presum - k] != 0:
            count += preSums[presum - k]   # 利用defaultdict的特性，当presum-k不存在时，返回的是0。这样避免了判断

            preSums[presum] += 1  # 给前缀和为presum的个数加1
            
        return count