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

最新推荐文章于 2024-09-20 21:16:31 发布

卖菜的小白

最新推荐文章于 2024-09-20 21:16:31 发布

阅读量1k

点赞数 1

分类专栏：面试算法文章标签： leetcode js 算法面试

本文链接：https://blog.csdn.net/weixin_47450807/article/details/124220705

版权

面试算法专栏收录该内容

121 篇文章 4 订阅

订阅专栏

一、写在前面

给你一个整数数组 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
};