【问题标题】:Given an array of integers nums and an integer k, return the total number of continuous subarrays whose sum equals to k给定一个整数数组 nums 和一个整数 k,返回总和等于 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)

''' 这里当目标

【问题讨论】:

标签: python-3.x backtracking recursive-backtracking


【解决方案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

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2020-06-30
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多