【hot100】刷题记录(5)-和为K的数组(*)

题目描述：

给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的子数组的个数 。

子数组是数组中元素的连续非空序列。

 

示例 1：

输入：nums = [1,1,1], k = 2
输出：2

示例 2：

输入：nums = [1,2,3], k = 3
输出：2

 

提示：

• 1 <= nums.length <= 2 * 104
• -1000 <= nums[i] <= 1000
• -107 <= k <= 107

 

我的作答：

第一次写。。忽略了nums[i]会小于0.。。。。。然后暴力解法果不其然超时了。我还在傻傻地用滑动窗口，笑死

看了题解，觉得前缀和这个概念很有必要牢记于心。。。。。。意思是每遍历一个元素nums[i]，就把这个累积到这个元素的总和pre[i]用一个字典收集起来{pre[i]:出现次数}，而目标值就是pre[i]-k是否在字典里。

太难想到了呜呜 o（n）

class Solution(object):
    def subarraySum(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        count, pre = 0, 0
        mydict = {}
        mydict[0] = 1
        for i in range(len(nums)):
            pre += nums[i]
            if pre-k in mydict:
                count += mydict[pre-k]
            mydict[pre] = mydict.get(pre, 0)+1
        return count

 

参考：

差不多