【Golang】LeetCode 560. 和为 K 的子数组

YGGP2026-02-15 8:50

560. 和为 K 的子数组

题目描述

思路

看到这道题目我们可能会想到使用滑动窗口来解题：设置两个指针来对窗口进行维护，不断地移动右指针并将右指针指向的数值累加到sum当中，如果sum > k，就开始移动左指针并减去左指针对应的数的数值。但是这个思路是错误的，原因就是数组当中可能存在负数，基于窗口的和与目标数值比大小没有意义。

正确的做法应该是基于前缀和来判断数组中是否有连续的子数组和为k，我们只需要记录每个前缀和出现的频率即可，我们设置一个字典mp来对前缀和出现的频率进行统计，用prefix来在遍历数组的过程中对前缀和进行维护。如果prefix - k在mp当中出现过，就代表着当前数组中存在一个子数组，其和为k。由于数组当中可能存在负数，因此我们通过mp对前缀和出现的频率进行统计，这种情况下我们令答案ans与mp[prefix - k]进行累加即可。

基于以上思路，我们写代码来解决问题。

Golang 题解

go 复制代码

func subarraySum(nums []int, k int) int {
    mp := map[int]int{}
    mp[0] = 1

    n, prefix, ans := len(nums), 0, 0
    for i := 0; i < n; i ++ {
        prefix += nums[i]
        if cnt, ok := mp[prefix - k]; ok {
            ans += cnt
        }
        mp[prefix] += 1
    }

    return ans
}

Python 题解

python 复制代码

class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
        mp = dict()
        mp[0] = 1

        n, prefix, ans = len(nums), 0, 0
        for num in nums:
            prefix += num
            if prefix - k in mp:
                ans += mp[prefix - k]

            if prefix not in mp:
                mp[prefix] = 0

            mp[prefix] += 1
        
        return ans