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
解题思路:
用哈希表存储前缀和,cunrrent_sum = current_sum + num,判断target = current_sum-k是否存在与之前的前缀和中举个例子,pre[1] = -7, pre[5]=-7,代表当前的子数中有两种方式来得到-7,一种是[0, 1]另一种是[0, 1, 2, 3, 4, 5],当target = -7时代表当前的num需要+7才能得到k这就有两种方式,一种是0, 1....i而另一种就是0, 1...5...i,换句话说0-i和2-i得到的结果是相同的
代码:
class Solution:
def subarraySum(self, nums, k: int) -> int:
hash_map = {0:1}
sum = 0
count = 0
for num in nums:
sum +=num
target = sum-k
if target in hash_map:
count += hash_map[target]
hash_map[sum] = hash_map.get(sum, 0) +1
return count