560. 和为 K 的子数组 中等

作者:Choi Yang

更新于:7 个月前

字数统计:201 字

阅读时长:1 分钟

题目描述

Tags

array | hash-table

Companies

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

解题思路

/**
 * @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
}

Contributors

Choi Yang