描述
给定一个整数数组和一个整数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
}