给定一个整数数组 nums 和一个整数 k，返回总和等于 k ​​的连续子数组的总数

Posted 2023-02-22

【中文标题】给定一个整数数组 nums 和一个整数 k，返回总和等于 k ​​的连续子数组的总数

【英文标题】：Given an array of integers nums and an integer k, return the total number of continuous subarrays whose sum equals to k

【发布时间】：2021-01-27 08:45:25

【问题描述】：

def subarraySum(self, nums: List[int], k: int) -> int:
    count = 0
    target = k
    self.cal(nums,target,count,k)
    return count
def cal(nums,target, count,k):
    if target == 0:
        count = count+1
        target = k
        return count,target
    if target<0:
        return 
        
    for i in range(len(nums)):
        self.cal(nums[:i]+nums[i+1:],target-nums[i],count,k)

''' 这里当目标

【问题讨论】：

另一个 Leetcode 问题？为算法添加标签会有所帮助。那么你的问题是什么？ （你应该原始源链接）

这是 Leetcode #560 - leetcode.com/problems/subarray-sum-equals-k

【参考方案1】：

您可以尝试使用prefix 思想和defaultdict() 更优雅有效地解决这个子数组求和问题。这取决于prefix sum 的想法，代码很容易理解，您可以尝试使用不同的数据运行它。在dc[v] 中，我们存储所有前缀和，即数字。上一个以值 v 为总和前缀。然后它循环和数组以查看是否有新的 num。 wcoming 的值满足w-v equal k 然后我们得到一个新的计数（对）。

from collections import defaultdict
class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
        dc = defaultdict(int)
        sum_ = 0
        dc[0] = 1
        result = 0
        for n in nums:
            sum_ += n
            if sum_ - k in dc:
                result += dc[sum_ - k]
            dc[sum_] += 1
        return result