Leetcode刷题(第560题)——和为 K 的子数组

一、写在前面

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

二、示例

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

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

三、思路

本题采用前缀树，只要求得两个前缀数之差为k即可。

四、代码展示

var subarraySum = function (nums, k) {
  let len = nums.length
  let preNumsTotal = new Array(len)
  preNumsTotal[0] = nums[0]
  for (let i = 1; i < len; i++) {
    preNumsTotal[i] = preNumsTotal[i - 1] + nums[i]
  }
  let map = new Map()
  map.set(0, 1)
  let count = 0
  preNumsTotal.forEach(item => {
    if (map.has(item - k)) {
      count += map.get(item - k)
    }
    if(map.has(item)) {
      map.set(item, map.get(item) + 1)
    }else {
      map.set(item, 1)
    }
  })
  return count
};

五、总结