leetcode 560. 和为K的子数组

题目描述

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

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

说明 :

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

题解

通过遍历nums,记录连续数值的和sum出现的次数,sum-k表示当前和与前边某一个和相差k,也就是说这两个和值之间的子数组就是和为K的子数组

示例代码(go)

1
2
3
4
5
6
7
8
9
10
11
func subarraySum(nums []int, k int) int {
res, sum := 0, 0
hash := make(map[int]int, len(nums))
hash[0] = 1
for _, v := range nums {
sum += v
res += hash[sum-k]
hash[sum]++
}
return res
}