LeetCode - 560. 和为K的子数组（python）

最新推荐文章于 2025-06-22 15:54:51 发布

原创最新推荐文章于 2025-06-22 15:54:51 发布 · 754 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#python #leetcode #数据结构 #算法

实习同时被 3 个专栏收录

42 篇文章

订阅专栏

python

41 篇文章

订阅专栏

leetcode

16 篇文章

订阅专栏

给定一个整数数组和目标值k，本篇博客探讨如何找出数组中和为k的连续子数组的个数。例如，在数组[1,1,1]和目标k=2的情况下，存在两个符合条件的子数组[1,1]。解题关键在于利用前缀和与字典来统计子数组出现的次数，注意处理特殊情况，如数组元素可能小于0，以及单个元素等于k的情况。" 112292484,10296227,面试技巧：如何回答‘工作中遇到的难题及解决方法’,"['面试策略', '面试问题', '求职技巧', '技术面试', '职业发展']

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

560. 和为K的子数组

给定一个整数数组和一个整数 k，你需要找到该数组中和为 k 的连续的子数组的个数。

示例 1 :
输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。

说明 :
数组的长度为 [1, 20,000]。
数组中元素的范围是 [-1000, 1000] ，且整数 k 的范围是 [-1e7, 1e7]。

解题：

# 一开始采用暴力解法，超出了时间限制
class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
        result=0
        for i in range(len(nums)):
            s=nums[i]
            if s==k:
                result+=1
            for j in range(i+1,len(nums)):
                s+=nums[j]
                if s==k:
                    result+=1
        return result
# 改为如下代码
def subarraySum(nums, k):
    dic = {}
    # 注意此处如果没有dic[0]=1的话，后面必须要有对单个元素等于k的判断；如果有判断的话则不需要dic[0]=1
    # dic[0]=1
    acc, res = 0, 0
    for num in nums:
        acc += num
        if acc == k:
            res += 1
        if acc - k in dic:
            res += dic[acc - k]
        dic[acc] = dic.get(acc, 0) + 1
    return res

总结：

注意数组元素并不都大于0。
注意只有一个元素就满足k的情况。
dict.get(key, default=None)，参数：key – 字典中要查找的键。default – 如果指定键的值不存在时，返回该默认值。
字典：dic={}。
前缀和：该元素位置之前所有元素的总和。
i到j之间子数组总和为k，则nums[j]-nums[i-1]=k，则有nums[i-1]=nums[j]-k，所以判断当前元素与k之间的差值是否是前若干元素总和即可，注意计数时并不都是+1，如果前若干元素总和=nums[j]-k的子数组个数大于1，则要全部加上，因为是不同的子数组。