560. 和为K的子数组

在这里插入图片描述
有负数,不能用滑动窗口了。考虑前缀和思想,用一个map记录从0到i的每个和是否出现过,用让res+=ma[当前和-k],为什么是+=,而不是判断有无这个键然后+1呢,因为有负数啊!!我们需要的那个前缀和可能多次出现了,所以res+上所有从0到i出现这个前缀和的个数。
举个例子,我们有1,2,3,然后k是5,累加和最后到6,所以我们只要减去一段和为1的前缀和即可,用ma判断和为1的前缀和有几种情况,全让res加上

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        //不是滑动窗口,要考虑负数,不能收缩left
        //如果是二维dp也有点麻烦,就是记录从几到几
        //vector<vector<int>> dp;
        //好像二维dp和普通的暴力法一样...
        //好像还是有滑动窗口那个味的,滑动窗口O(N)在于left一次能走很多步

        //前缀和,不是普通的前缀和,因为前缀和也要O(N^2)的时复
        //相当于和为0的和已经出现过一次了,也就是一个数字都不取
        unordered_map<int,int> ma{{0,1}};
        int res = 0;
        int sum = 0;//记录从0到i的和为多少,然后判断sum-k的前缀和有几个
        //意思就是123,我们要求和为5的,就一路加,1,3,6,最后一次判断ma[6-5]不为0,也就是有一段从0到某一个数字x的区间值为1,那么从x到当前数字的和就是5
        for(int x:nums){
            sum += x;
            res += ma[sum-k];
            ma[sum]++;
        }
        return res;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值