leetcode-剑指 Offer II 010. 和为 k 的子数组

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

 

示例 1 :

输入:nums = [1,1,1], k = 2
输出: 2
解释: 此题 [1,1] 与 [1,1] 为两种不同的情况

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/QTMn0o
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

1. 简单思路 【超时】

暴力遍历
从当前位置开始向后遍历只要满足和为 k 就将结果加1

class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
        cursor = 0
        res = 0
        while cursor < len(nums):
            i = cursor
            curr = 0
            while i < len(nums):
                curr += nums[i]
                if curr == k:
                    res += 1
                i += 1
                
            cursor += 1
        return res

2. 前缀和

任意子序列的和都可以通过 [0,i] [0,j] 两个子序列相减得到 因此从0开始计算 [0, 0]
[0,1]…[0,len -1]的和 如果差值为 k 则将结果加1

class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
        sub_array_sum_map = {}
        sum = 0
        result = 0
        # 处理边界情况 
        sub_array_sum_map[0] = 1
        for i in range(0, len(nums)):
            sum += nums[i]
            result += self.get(sub_array_sum_map, sum - k, 0)
            sub_array_sum_map[sum] = self.get(sub_array_sum_map, sum, 0) + 1
        return result

    def get(self, dic, key, default):
        if dic.get(key) is None:
            return default
        return dic.get(key)