【LeetCode】和为 K 的子数组

---

一、题目

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

子数组是数组中元素的连续非空序列。

示例 1：

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

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

提示：

1 <= nums.length <= 2 * 10 ^4
-1000 <= nums[i] <= 1000
-10 ^7 <= k <= 10 ^7

---

二、解法

前缀和思想
s(i...j)的和SUM(s(i...j)) = SUM(s(0...j)) - SUM(s(0...i-1))
因此我们可以把遇到的前缀和放到一个哈希表里，key存储前缀和本身，value存出现的次数
注意，为了同一操作，前缀和为0 出现的次数为1，类似于dummy头节点

---

完整代码

class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
        res = 0
        dic = {}
        dic[0] = 1
        pre = 0
        for x in nums:
            pre += x
            if pre - k in dic:
                res += dic[pre - k]
            dic[pre] = dic.get(pre, 0) + 1
        return res