思路
本题的主要思路为创建一个哈希表记录每个0~i的和,在遍历这个数组的时候查询有没有sum-k的值在哈希表中,如果有,说明有个位置到当前位置的和为k。
有可能不止一个,哈希表负责记录有几个sum-k,将和记录下来。这里把每个以i结尾的结果都算上了,所以不用担心不全等问题。
这里需要注意的两个点为:
- m[0]=1;提前在哈希表注册一个0的结果,因为如果刚好是0到当前i的话,需要有一个记录被算上。
- m[sum]++这行代码放的位置,有个简单的方法来判断会不会影响当前i计算的结果,那就是加到当前的和为0,k也为0,如果你放在前面的话就会算上这个结果,但是当前i到当前i是没有元素的,不算一种结果,但是你加上了这个结果,是错误的,有影响的。
代码示例
func subarraySum(nums []int, k int) int {
result:=0
m:=make(map[int]int)
m[0]=1
sum:=0
for i:=0;i<len(nums);i++{
sum+=nums[i]
if val,ok:=m[sum-k];ok{
result+=val
}
m[sum]++
}
return result
}