给你一个整数数组 nums
和一个整数 k
,请你统计并返回 该数组中和为 k
的连续子数组的个数 。
示例 1:
输入:nums = [1,1,1], k = 2 输出:2
示例 2:
输入:nums = [1,2,3], k = 3 输出:2
思路:
遍历,计算从索引0开始到当前位置的所有子串和total,并将该和的出现次数存入哈希表count中,在遍历过程中,若子串和为k,数量res + 1,若当前子串和total减去k在哈希表中出现过,数量res + 该差值出现的次数。
如果total - k在哈希表中出现过意味着什么 比如前面几个数字相加和为x,再加上后面几个数以后和变成x + k = total,说明从和为x的数到当前数之间的数和为k,所以相应的次数res要增加,此时的情况就是 total - k = x 这个x在之前已出现过,且出现过几次res就增加几。
class Solution:
def subarraySum(self, nums: List[int], k: int) -> int:
total = 0
res = 0
count = {}
for num in nums:
total += num
if total == k:
res += 1
if total - k in count:
res += count[total - k]
count[total] = count.get(total, 0) + 1#记录每个total出现的次数
return res