560.和为k的子数组

题目:

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

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

示例 1:

输入:nums = [1,1,1], k = 2
输出:2

示例 2:

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

提示:

  • 1 <= nums.length <= 2 * 104
  • -1000 <= nums[i] <= 1000
  • -107 <= k <= 107

解题思路:

用哈希表存储前缀和,cunrrent_sum = current_sum + num,判断target = current_sum-k是否存在与之前的前缀和中举个例子,pre[1] = -7, pre[5]=-7,代表当前的子数中有两种方式来得到-7,一种是[0, 1]另一种是[0, 1, 2, 3, 4, 5],当target = -7时代表当前的num需要+7才能得到k这就有两种方式,一种是0, 1....i而另一种就是0, 1...5...i,换句话说0-i和2-i得到的结果是相同的

代码:

class Solution:
    def subarraySum(self, nums, k: int) -> int:
        hash_map = {0:1}
        sum = 0
        count = 0
        for num in nums:
            sum +=num
            target = sum-k
            if target in hash_map:
                count += hash_map[target]
            hash_map[sum] = hash_map.get(sum, 0) +1
        return count

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值