文章作者:Tyan
博客:noahsnail.com | CSDN | 简书
1. Description
2. Solution
**解析:**Version 1,使用前缀和来解决,遍历数组,求前缀和,统计前缀和的次数并保存到字典中,当碰到差值在字典中存在时,则意味着当前数组减去之前的前缀和数组等于k
,将次数加到count
中,更新前缀和的次数。注意,假设第一个数就等于k
,此时数组中没有差值0
的次数,因此stat[0]=1
。
- Version 1
class Solution:
def subarraySum(self, nums: List[int], k: int) -> int:
n = len(nums)
total = 0
count = 0
stat = collections.defaultdict(int)
stat[0] = 1
for i in range(n):
total += nums[i]
count += stat[total - k]
stat[total] += 1
return count