子阵列总和等于 K

Question

给定一个整数数组和一个整数 k，您需要找到总和等于 k ​​的连续子数组的总数。

试图从leetcode解决这个问题，但不确定我到底卡在哪里以及如何找到正确的解决方案。 对于 nums = [0,0,0,0,0,0,0,0,0,0] 和 k = 0 的情况，它失败了，我得到的答案是 10，而实际上应该是 55

class Solution(object):
    def subarraySum(self, nums, k):
        count = 0
        for i in range(0, len(nums)): 
            count += self.driver(nums, k, nums[i], i+1)
        return count


    def driver(self, nums, target, curr, index): 
        if curr == target:
            return 1
        else:
            for i in range(index, len(nums)): 
                return self.driver(nums, target, curr + nums[i], i+1)
            return 0

Answer 1

使用 itertools 的累积功能会更容易。

from itertools import accumulate
def subarraySum(a,k):
    return sum( list(accumulate(a[i:])).count(k) for i in range(len(a)) )

print(subarraySum([0,0,0,0,0,0,0,0,0,0],0)) # 55

只需让accumulate() 计算从数组中每个位置到最后的累积总和，并计算它到达所需值的次数。 这将涵盖零和负数的情况。

如果不允许使用导入的模块，则可以编写自己的累积迭代器：

def accumulate(a):
    s = 0
    for v in a: s+=v; yield s