10.和为 K 的子数组(学习)
给你一个整数数组 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
var subarraySum = function(nums, k) {
let prefixSumCount = { 0: 1 }; // 初始化前缀和为0的次数为1
let currentSum = 0;
let count = 0;
for (let num of nums) {
currentSum += num;
let target = currentSum - k;
if (prefixSumCount[target] !== undefined) {
// 如果存在target作为前缀和,则找到了一个满足条件的子数组
count += prefixSumCount[target];
}
// 更新当前前缀和的次数
if (prefixSumCount[currentSum] === undefined) {
prefixSumCount[currentSum] = 1;
} else {
prefixSumCount[currentSum]++;
}
}
return count;
};