和为 K 的子数组

Problem: 560. 和为 K 的子数组
给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的子数组的个数 。

子数组是数组中元素的连续非空序列。

示例 1：

输入：nums = [1,1,1], k = 2
输出：2
示例 2：

输入：nums = [1,2,3], k = 3
输出：2

思路

看了提示 使用前缀和配合哈希

Code

func subarraySum(nums []int, k int) int {
	result := 0
	// 看了提示 使用前缀和配合哈希
	// 使用循环计算前缀和
	prefixSum := 0
	mapIntInt := make(map[int]int)
	mapIntInt[0]++
	for _, value := range nums {
		// 当前前缀和减去k即为需要查找的前缀和
		prefixSum += value
		if v, ok := mapIntInt[prefixSum-k]; ok {
			// 找到
			result += v
		}
		// 将当前前缀和存入map
		mapIntInt[prefixSum]++
	}
	return result
}