560. 和为 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

 

 

class Solution(object):

    def subarraySum(self, nums, k):

        """

        :type nums: List[int]

        :type k: int

        :rtype: int

        """

        count = 0

        n = len(nums)

        preSums = collections.defaultdict(int)

        preSums[0] = 1

        presum = 0

        for i in range(n):

            presum += nums[i]

            count += preSums[presum - k]

            preSums[presum] += 1

        return count