给定一个整数数组和一个整数 k ,请找到该数组中和为 k 的连续子数组的个数。
示例 1 :
输入:nums = [1,1,1], k = 2
输出: 2
解释: 此题 [1,1] 与 [1,1] 为两种不同的情况
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/QTMn0o
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
文章目录
1. 简单思路 【超时】
暴力遍历
从当前位置开始向后遍历只要满足和为 k 就将结果加1
class Solution:
def subarraySum(self, nums: List[int], k: int) -> int:
cursor = 0
res = 0
while cursor < len(nums):
i = cursor
curr = 0
while i < len(nums):
curr += nums[i]
if curr == k:
res += 1
i += 1
cursor += 1
return res
2. 前缀和
任意子序列的和都可以通过 [0,i] [0,j] 两个子序列相减得到 因此从0开始计算 [0, 0]
[0,1]…[0,len -1]的和 如果差值为 k 则将结果加1
class Solution:
def subarraySum(self, nums: List[int], k: int) -> int:
sub_array_sum_map = {}
sum = 0
result = 0
# 处理边界情况
sub_array_sum_map[0] = 1
for i in range(0, len(nums)):
sum += nums[i]
result += self.get(sub_array_sum_map, sum - k, 0)
sub_array_sum_map[sum] = self.get(sub_array_sum_map, sum, 0) + 1
return result
def get(self, dic, key, default):
if dic.get(key) is None:
return default
return dic.get(key)