Bootstrap

一起学习LeetCode热题100道(10/100)

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;  
};
;