560. 和为 K 的子数组

思路

  本题的主要思路为创建一个哈希表记录每个0~i的和,在遍历这个数组的时候查询有没有sum-k的值在哈希表中,如果有,说明有个位置到当前位置的和为k。
560. 和为 K 的子数组_第1张图片
  有可能不止一个,哈希表负责记录有几个sum-k,将和记录下来。这里把每个以i结尾的结果都算上了,所以不用担心不全等问题。
  这里需要注意的两个点为:

  1. m[0]=1;提前在哈希表注册一个0的结果,因为如果刚好是0到当前i的话,需要有一个记录被算上。
  2. 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
}

你可能感兴趣的:(算法,数据结构)