【leetcode系列】【算法】【中等】和为K的子数组

题目：

题目链接： https://leetcode-cn.com/problems/subarray-sum-equals-k/

 

解题思路：

有点类似于“两数之和”的思路

遍历数组的时候，同时累加从第一个元素到当前元素的和sum，放入到hash表中。同时判断sum - k是否在hash表中存在

如果存在，说明之前至少存在一个下标为idx的元素，idx之前的子数组之和为sum - k，idx之后的子数组适合为k，这个时候更新结果个数 += 子数组之和为sum - k的个数

 

代码实现：

class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
        rec = collections.defaultdict(int)
        # 如果不把0加进去，如果遍历时的当前元素 = k，则无法找到
        rec[0] = 1
        curr_sum = 0
        res = 0
        for a in nums:
            # 对和进行累加(如果数组特别大,这里可能需要考虑值类型溢出的事情)
            curr_sum += a
            # 如果curr_sum - k在rec表中存在，说明至少存在一个下标为idx的元素
            # idx之前的子数组之和为curr_sum - k，idx到当前元素的子数组值和为k
            if curr_sum - k in rec:
                res += rec[curr_sum - k]

            rec[curr_sum] += 1

        return res