LeetCode 热题 100——子串——和为 K 的子数组

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

求解

（1）暴力求解
遍历所有可能。一开始，还想着排序，不过排序就乱了。。。

var subarraySum = function(nums, k) {if (nums.length === 0) return 0let ans = 0;for (let i = 0; i < nums.length; i++) {// 单个数字if (nums[i] === k) ans++let item = nums[i]for (let j = i + 1; j < nums.length; j++) {item += nums[j]if (item === k) {ans ++}}}return ans
};

（2）优化：前缀和 + 哈希表

• 什么是前缀和？
  定义一个前缀和数组 prefixSum，其中 prefixSum[i] 表示数组 nums 中前 i 个元素的和（即 nums[0] 到 nums[i-1] 的和）

  prefixSum[0] = 0 (前 0 个元素的和，也就是没有元素的和)
  prefixSum[1] = nums[0]
  prefixSum[2] = nums[0] + nums[1]
  prefixSum[3] = nums[0] + nums[1] + nums[2]
  ...
  prefixSum[n] = nums[0] + nums[1] + ... + nums[n-1]
  

  那么，数组中从索引 j 到 i-1 的子数组（即 nums[j], nums[j+1], ..., nums[i-1]）的和就可以表示为：sum(nums[j..i-1]) = prefixSum[i] - prefixSum[j]

• 问题转化
  我们的目标是找到所有满足 sum(nums[j..i-1]) = k 的子数组。根据上面的公式，这等价于找到所有满足 prefixSum[i] - prefixSum[j] = k 的 (i, j) 对。
  我们可以将其变形为：prefixSum[j] = prefixSum[i] - k。
  思路就变成了：当我们遍历到第 i 个元素，计算出 prefixSum[i] 时，我们想知道在 i 之前，有多少个 j 满足 prefixSum[j] 的值等于 prefixSum[i] - k。每找到一个这样的 j，就意味着存在一个以当前 i-1 为结尾的子数组，其和为 k。

• 哈希表的作用
  为了高效地回答 “之前有多少个 prefixSum[j] 等于某个值” 这个问题，我们可以使用一个哈希表（在 JavaScript 中是 Map 或普通对象）。

  • 键 (Key)：我们遍历过程中遇到的 prefixSum 值。
  • 值 (Value)：该 prefixSum 值出现的次数。

具体代码：

var subarraySum = function(nums, k) {if (nums.length === 0) return 0let ans = 0let preSum = 0let map = new Map() // 存放 前缀和 出现的次数map.set(0, 1) // 空集for (let num of nums) {preSum += num // 求前缀和// 判断 是否有 符合条件的，有则将前缀和加起来if (map.has(preSum - k)) ans += map.get(preSum - k)// 将当前前缀和加入map.set(preSum, (map.get(preSum) || 0) + 1)}return ans
};