Skip to content

560. 和为 K 的子数组 中等

作者:Choi Yang
更新于:6 个月前
字数统计: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

解题思路

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