leetcode 27和为k的子数组

最新推荐文章于 2024-06-15 09:23:13 发布

RickyGuo00

最新推荐文章于 2024-06-15 09:23:13 发布

阅读量277

点赞数

文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/RickyGuo00/article/details/125923441

版权

day27笔记

- 1.题目描述
- 2.代码构思（debug）

1.题目描述

在这里插入图片描述

2.代码构思（debug）

在这里插入图片描述

class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
        if not nums:
            return
        
        
        # 要求的连续子数组
        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