Leetcode 560. 和为 K 的子数组

给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的子数组的个数 。
子数组是数组中元素的连续非空序列。
示例 1：
输入：nums = [1,1,1], k = 2
输出：2
示例 2：
输入：nums = [1,2,3], k = 3
输出：2

解题思路
前缀和+字典
暴力枚举的复杂度至少为O(n^2)，因此考虑使用前缀和。
假设nums的前缀和数组为sums，问题转化为：求使得sums[j]−sums[i]=k的下标对(i,j)个数，其中i<j。
利用字典存储前缀和以及对应个数。枚举前缀和数组，当枚举到sums[j]时，假设有键值对sums[j]−k:v,可知存在v个值满足要求，把v加入最终答案即可，并将sums[i]更新到字典当中。
最后，考虑空间优化，不需要前缀和数组。在计算当前前缀和的同时，完成更新最终答案和更新字典的工作。
python：

class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
        # 如果出现前缀和sums-k=0，则表明该前缀和即为1个满足条件的子数组，所以dic.get(0,0)不应该返回0，而是1
        dic={0:1}
        # 前缀和sums,指的是前n个数值的和
        sums=0
        # 满足条件的子数组的个数
        res=0
        for num in nums:
            sums+=num
            # dic.get(key,default)为查询dic中key的value，查不到时返回default
            res+=dic.get(sums-k,0)
            dic[sums]=dic.get(sums,0)+1
        return res