leetcode560. 和为 K 的子数组

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

示例 1:

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

示例 2:

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

思路:

        遍历,计算从索引0开始到当前位置的所有子串和total,并将该和的出现次数存入哈希表count中,在遍历过程中,若子串和为k,数量res + 1,若当前子串和total减去k在哈希表中出现过,数量res + 该差值出现的次数。
        如果total - k在哈希表中出现过意味着什么 比如前面几个数字相加和为x,再加上后面几个数以后和变成x + k = total,说明从和为x的数到当前数之间的数和为k,所以相应的次数res要增加,此时的情况就是 total - k = x 这个x在之前已出现过,且出现过几次res就增加几。

class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
        total = 0
        res = 0
        count = {}
        for num in nums:
            total += num
            if total == k:
                res += 1
            if total - k in count:
                res += count[total - k]
            count[total] = count.get(total, 0) + 1#记录每个total出现的次数
        return res

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

小梁今天敲代码了吗

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值