正文
一般解法
class Solution:
def subarraySum(self, nums: List[int], k: int) -> int:
number = 0
for i in range(len(nums)):
for j in range(i , len(nums)):
if sum(nums[i:j + 1]) == k:
number += 1
return number
上述方法虽然可行,但是时间复杂度为 O ( n 2 ) O\left(n^2\right) O(n2)。不够迅速,因此,我们考虑使用其他方式实现。
字典方法
import collections
from typing import List
class Solution:
def subarraySum(self, nums: List[int], k: int) -> int:
count = 0
dict1 = collections.defaultdict(int)
dict1[0] = 1
pre_sum = 0
for i in range(len(nums)):
pre_sum += nums[i]
count += dict1[pre_sum - k]
dict1[pre_sum] += 1
return count
需要特别注意的是,这里的 count += dict1[pre_sum - k]
和 dict1[pre_sum] += 1
位置不能互换。
如果大家觉得有用,就请点个赞吧~