560. 和为 K 的子数组 【@我的刷题日记】

难度：中等                日期：2023年5月15日

给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的连续子数组的个数 。

示例 1：

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

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

 方法：前缀和

思路：将所有前缀和都以值 和 出现的次数 的方式存储在map对象中，当存在 当前前缀和 减 原来的某一前缀和 等于 k 时，就表明存在字串的和 等于 k。查找的方法为：map[当前前缀和 - k] 如果存在则count++。

var subarraySum = function(nums, k) {
    let count = 0;
    //为了让第一个值也满足条件
    let map = {0:1};
    let currentSum = 0;//当前前缀和
    for(num of nums){
        currentSum += num; 
        if(map[currentSum-k]){ //如果存在 当前前缀和 - 某一前缀和 = k
            count += map[currentSum-k];
        }
        if(map[currentSum]){ //如果存过当前前缀和则++
            map[currentSum]++;
        }else{
            map[currentSum] = 1; //否则初始化为1
        }
    }
    return count;
};

结果