和为K的子数组（前缀和 + HashMap解法）python，golang

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

示例 1 :

输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。
思路：
我我们定义pre[i] 为 [0…i] 里所有数的和，则pre[i] 可以由 pre[i−1] 递推而来，即

pre[i]=pre[i−1]+nums[i]

那么「[j…i] 这个子数组和为 k，这个条件我们可以转化为

pre[i]−pre[j−1]==k

简单移项可得符合条件的下标 j 需要满足

pre[j−1]==pre[i]−k
这里直接引用该图表达上述式子的含义，presum即pre【j-1】，presum即pre【i】。

以上部分转载自力扣题解
Python解法

class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
       
        count = 0
        cur_sub_sum=0
        lookup={0:1}
        for num in nums:
            cur_sub_sum +=num
            if cur_sub_sum-k in lookup:
                count+=lookup[cur_sub_sum-k]
            lookup[cur_sub_sum]=lookup.get(cur_sub_sum,0)+1  
            #如果cur_sub_sum不在，则设置为0然后+1。若存在直接+1
        
        return count

Go解法

func subarraySum(nums []int, k int) int {
    count,pre:=0,0
    m:=map[int]int{}
    m[0]=1
    for i := range nums {
        pre += nums[i]
        if _, ok := m[pre - k]; ok {
            count += m[pre - k]
        }
        m[pre] += 1
    }
    return count
}