560. Subarray Sum Equals K

最新推荐文章于 2020-07-21 10:35:30 发布

原创最新推荐文章于 2020-07-21 10:35:30 发布 · 621 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

31 篇文章

订阅专栏

本文介绍了一种解决子数组和等于特定值K的问题的方法。通过使用哈希表记录前缀和出现的次数，可以有效地找出所有连续子数组中和为K的个数。这种方法避免了暴力枚举所有可能子序列的低效方式。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Description:
Given an array of integers and an integer k, you need to find the total number of continuous subarrays whose sum equals to k.

Solution:

暴力列举所有可能的子连续序列和
计算前n个序列的和，并储存与map中，再根据nums[0:i] - nums[0:j]的值来计算nums[j:i]的和

class Solution(object):
    def subarraySum(self, nums, k):
        sums, cnt, dict1 = 0, 0, {0:1}
        for i in nums:
            sums += i
            if dict1.get(sums - k) != None:
                cnt += dict1[sums - k]
            dict1[sums] = 1 if dict1.get(sums) == None else dict1[sums] + 1

        return cnt