在做题中学习(58):和为K的子数组

560. 和为 K 的子数组 - 力扣(LeetCode)

因为是判断子数组的和 + 要返回 == k  的次数,所以

解法:前缀和 + 哈希表

提出一个概念:以下标i为结尾的所有子数组

那要找出所有和 == k的子数组 就相当于:找出所有值为dp[i] - k的子数组。

 dp[i] - k就是一段区间的前置和了,只要把所有以下标i为结尾的所有子数组的前缀和丢到哈希表中,最后返回 dp[i] - k (前提:要存在)出现的次数即可。

细节

1.前缀和什么时候丢入哈希表?

因为放进去的是前缀和,放[0,i]的话,到了[0,i+1]就会重复,所以到i下标时,先放入[0,i-1]的前缀和,再加入自己。(dp[i])

2.不用真的创建一个前缀和数组

dp[i] = sum + nums[i]  ,所以只需要一个sum代表前一次的前缀和,每次算完前缀和,把sum更新就行了。

3.如果到自己的前缀和是k呢?(dp[i] == k)

那范围就是:[0,-1],显然不行,所以默认让hash[0] = 1

class Solution 
{
public:
    int subarraySum(vector<int>& nums, int k) 
    {
        unordered_map<int,int> hash;
        hash[0] = 1;
        int sum = 0,count = 0;
        for(auto e : nums)
        {
            sum += e;
            //判断操作就是在存自己之前,判断前面的前缀和
            if(hash.count(sum-k))
                count += hash[sum - k];
            //存完[0,i-1]后,再存自己
            hash[sum]++;

        }
        return count;


    }
};

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值