前缀和--和为k的数组 560

求一个数组中连续数字的和为k的子数组个数。
一开始想到是滑动窗口,但是滑动窗口并不能解决,因为有负数,所以走过的还可能要用到。
又想到动态规划,也不行,因为状态转移式推不出来。
可以利用前缀和,前缀和就是用来解决连续子数组的和的问题,可以快速得到某个子数组的和,用一个前缀和数组来对原数组进行预处理。
若定义pre(i)是0到i的所有数的和,则一个j到i的子数组的和就可以用pre(j)和pre(i)来表示。
这个题的意思就是求多少个和为k的连续子数组,则最直接的想法是就看以i为结尾的数组中,有多少个是满足他们的前缀和是等于k的。但这样是O(n2)
所以如果想要改进,改一下表达式pre(j)=pre(i)-k时成立
可以用一个map来在存前缀和的同时,存下该前缀和出现的次数,所以在遍历到i的时候要找有几个pre(i)-k的前缀和时可以直接得到次数,就不用for循环遍历找了。

public class Solution {
    public int subarraySum(int[] nums, int k) {
        int count = 0, pre = 0;
        HashMap <Integer, Integer> map = new HashMap <Integer, Integer>();//存的是此时的pre值对应的次数
        map.put(0, 1);//当pre为0时次数为1
        for (int i = 0; i < nums.length; i++) {//求每个i有几个pre[i]-k
            pre += nums[i];//算此时i的pre数组
            if (map.containsKey(pre - k)) {//如果拥有则加到次数中
                count+=map.get(pre - k);
            }
            map.put(pre, map.getOrDefault(pre, 0) + 1);//把此时的pre次数更新到map中
        }
        return count;
    }
}

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值