leetcode 560. 和为K的子数组 golang实现

描述

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

示例 1 :

输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。
说明 :

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

思路

我们记录每个索引对应的sum i < j 则 i+1到j的和 = sum[j] - sum[i]

实现

func subarraySum(nums []int, k int) int {
	var sumMap = make(map[int]int)
	var sum int
	var count int

	sumMap[0] = 1
	for _, num := range nums{
		sum += num
		diff := sum - k
		if times, ok := sumMap[diff];ok{
			count += times
		}
		sumMap[sum]++ // 本语句要放到if计数后面 否则会出现k=0时 每次把当前遍历的数组统计进去 比如[-1] k = 0 答案应该是0  如果先sumMap[sum]++ 统计时diff = -1 把nums[0:1]也统计进去了 实际上这个答案是不对的
	}

	return count
}