560.和为k 的子数组

·题目

给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的子数组的个数 。

子数组是数组中元素的连续非空序列。

 

示例 1：

输入：nums = [1,1,1], k = 2
输出：2
示例 2：

输入：nums = [1,2,3], k = 3
输出：2

·解题思路

首先是两个for循环暴力求解，然后导致验证超时。转为使用哈希表来记录出现的和的次数

————————暴力求解超时代码——————

class Solution(object):
    def subarraySum(self, nums, k):
        count = 0

        n = len(nums)
        for i in range(n):
            sub_sum = nums[i]
            if sub_sum == k:
                count += 1
            for j in range(i+1,n):
                sub_sum  += nums[j]
                if sub_sum == k:
                     count += 1

        return count

——————哈希表————————

class Solution(object):
    def subarraySum(self, nums, k):
        sum_map = {0:1}
        count = 0
        cur_sum = 0

        for num in nums:
            cur_sum += num
            if cur_sum - k in sum_map:
                count += sum_map[cur_sum - k ]

            sum_map[cur_sum] = sum_map.get(cur_sum,0) + 1

        return count