Leetcode—【中等】560. 和为K的子数组

题目：

给定一个整数数组和一个整数 k，你需要找到该数组中和为 k 的连续的子数组的个数。

示例 1 :

输入:nums = [1,1,1], k = 2

输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。

说明 :

数组的长度为 [1, 20,000]。

数组中元素的范围是 [-1000, 1000] ，且整数 k 的范围是 [-1e7, 1e7]。

解题思路：

（下面方法会超时）

class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
        cnt = 0
        for i in range(len(nums)):
            for j in range(i,len(nums)):
                if sum(nums[i:j+1])==k:
                    cnt +=1
        return cnt

那么接着看：

注意数组元素并不都大于0。

注意只有一个元素就满足k的情况。

dict.get(key, default=None)，参数：key – 字典中要查找的键。default – 如果指定键的值不存在时，返回该默认值。

字典：dic={}。

前缀和：该元素位置之前所有元素的总和。

i到j之间子数组总和为k，则nums[j]-nums[i-1]=k，则有nums[i-1]=nums[j]-k，所以判断当前元素与k之间的差值是否是前若干元素总和即可，注意计数时并不都是+1，如果前若干元素总和=nums[j]-k的子数组个数大于1，则要全部加上，因为是不同的子数组。

那么Python3代码如下：

class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
        prefixSumArray = {0:1}
        count = 0
        prefixSum = 0
        
        for ele in nums:
            prefixSum += ele
            subArray = prefixSum - k
            
            if subArray in prefixSumArray:
                count += prefixSumArray[subArray]
            '''
            prefixSumArray.get(prefixSum, 0)
            在hash table里查找key，如果有返回对应的value，反之返回0 
            '''
            prefixSumArray[prefixSum] = prefixSumArray.get(prefixSum, 0) + 1
        
        return count