560. [✔][M]和为 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

---

题解：

前缀和，寻找的子数组正好是当前前缀和-（preSum - k）前缀和，中间正好是k的连续数组。

// @lc code=start
function subarraySum(nums: number[], k: number): number {
    let map = new Map();
    let preSum = 0;
    let count = 0;

    map.set(0, 1);

    for (let i = 0; i < nums.length; i++) {
        preSum += nums[i];

        if (map.has(preSum - k)) {
            count += map.get(preSum - k);
        }

        if (map.has(preSum)) {
            map.set(preSum, map.get(preSum));
        } else {
            map.set(preSum, 1);
        }
    }

    return count;
};
// @lc code=end