Liaoyi560. 和为 K 的子数组 中等
题目描述
Tags
array | hash-tableCompanies
google给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的连续子数组的个数 。
示例 1：
输入：nums = [1,1,1], k = 2
输出：2
示例 2：
输入：nums = [1,2,3], k = 3
输出：2
提示：
- 1 <= nums.length <= 2 * 104
- -1000 <= nums[i] <= 1000
- -107 <= k <= 107
解题思路
js
/**
* @param {number[]} nums
* @param {number} k
* @return {number}
*/
const subarraySum = function (nums, k) {
const map = new Map()
map.set(0, 1)
let count = 0
let preSum = 0
for (const num of nums) {
// 前缀和
preSum += num
// 如果之前出现过preSum - k的前缀和，累加到count
count += map.get(preSum - k) || 0
// 存储当前前缀和出现的次数
map.set(preSum, (map.get(preSum) || 0) + 1)
}
return count
}