https://leetcode.cn/problems/subarray-sum-equals-k/
给你一个整数数组 nums 和一个整数 k ,请你统计并返回 该数组中和为 k 的子数组的个数 。
子数组是数组中元素的连续非空序列。
示例 1:
输入:nums = [1,1,1], k = 2
输出:2
示例 2:
输入:nums = [1,2,3], k = 3
输出:2
class Solution:
def subarraySum(self, nums: List[int], k: int) -> int:
# 数组、哈希表、前缀和
# 如果数组无负数,用滑窗,因为有可能取负数,所以用前缀和。
cnt, presum = 0, 0
# 如果第一次遍历遇到presum - k = 0,cnt要加上value,所以默认dic的value为1
dic = {0: 1}
for i in nums:
presum += i
# presum[j] - presum[i] = k -> presum[j] - k = presum[i]
# presum[j]减去k的结果如存在于前缀和数组,说明数组中存在和为 k 的连续子数组
cnt += dic.get(presum - k, 0)
dic[presum] = dic.get(presum, 0) + 1
return cnt