LeetCode.和为k的子数组(前缀和+哈希表)

题目描述:560. 和为 K 的子数组 - 力扣(LeetCode) (leetcode-cn.com)

 思路:前缀和:预处理求出s[1~n],然后就能快速求出任意数组任意区间里一段数的和,如求l~r区间的和,就是s[r]-s[l-1],但是如果求出前缀和再求任意子区间的和时间复杂度就是O(N*N),让求任意子区间的和是否为k,即i~j子区间s[j]-s[i]=k,那么我们可以换一个思路,找满足值为s[j]-k的s[i]的有多少个(因为s[j]-k如果等于s[i],则说明从i-1到j这一段子数组的和为k),即遍历所有以j结尾的前缀和,看看是否存在值为s[j]-k的值存在,但怎么判断是否有这样的前缀和存在呢,为了将时间复杂度缩小到O(N),可以借助一个哈希表,键值是以j结尾的区间的和,键对应的值是该区间和出现的次数,这样在找是否有这样的前缀和存在,存在多少个的复杂度就是O(1)。空间其实可以优化,我们并不需要一个前缀和数组,因为每次求的前缀和之和前一个前缀和有关,所以用一个pre变量来保存当前下标 j 的前缀和就可以,然后再哈希表里找,是否存在一个以 i 为下标的前缀和满足s[j]-k=s[i],如果存在,计数变量加上该前缀和存在的次数,并且每次都要将该次计算的前缀和存入哈希表。

代码实现C++:

int subarraySum(vector<int>& nums, int k) {
        int pre=0,count=0;
        unordered_map<int,int> mp;
        mp[0]=1;//初始化表示前缀和为0的出现次数为1
        for(auto num:nums){
            pre+=num;
            if(mp.find(pre-k)!=mp.end()){//如果查找满足条件的前缀和存在
                count+=mp[pre-k];//计数
            }
            mp[pre]++;//将前缀和存入哈希表
        }
        return count;
    }

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值