力扣
560. 和为K的子数组
给定一个整数数组和一个整数 k,你需要找到该数组中和为 k 的连续的子数组的个数。
示例 1 :
输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。
思路:
我我们定义pre[i] 为 [0…i] 里所有数的和,则pre[i] 可以由 pre[i−1] 递推而来,即
pre[i]=pre[i−1]+nums[i]
那么「[j…i] 这个子数组和为 k,这个条件我们可以转化为
pre[i]−pre[j−1]==k
简单移项可得符合条件的下标 j 需要满足
pre[j−1]==pre[i]−k
这里直接引用该图表达上述式子的含义,presum即pre【j-1】,presum即pre【i】。
以上部分转载自力扣题解
Python解法
class Solution:
def subarraySum(self, nums: List[int], k: int) -> int:
count = 0
cur_sub_sum=0
lookup={0:1}
for num in nums:
cur_sub_sum +=num
if cur_sub_sum-k in lookup:
count+=lookup[cur_sub_sum-k]
lookup[cur_sub_sum]=lookup.get(cur_sub_sum,0)+1
#如果cur_sub_sum不在,则设置为0然后+1。若存在直接+1
return count
Go解法
func subarraySum(nums []int, k int) int {
count,pre:=0,0
m:=map[int]int{}
m[0]=1
for i := range nums {
pre += nums[i]
if _, ok := m[pre - k]; ok {
count += m[pre - k]
}
m[pre] += 1
}
return count
}