Day04:剑指 Offer II 010. 和为 k 的子数组

题目

给定一个整数数组和一个整数 k ,请找到该数组中和为 k 的连续子数组的个数。

代码实现

因为有负数,所以我们不能使用滑动窗口

前缀和+双重遍历

效率很低

class Solution {
    public int subarraySum(int[] nums, int k) {
        int total = 0;
        int res = 0;
        for (int i = 1; i < nums.length; i++) {
            nums[i] = nums[i-1] + nums[i];
        }
        for (int i = 0; i < nums.length; i++) {
            if(nums[i] == k){
                res++;
            }
            for (int j = 0; j < i; j++) {
                if(nums[i] - nums[j] == k){
                    res++;
                }
            }
        }
        return res;
    }
}

前缀和+Map

大佬思路:
这道题目非常简洁,就是求数组中何为整数k的连续子数组个数。
如果这道题的取值没有负数,那就是标准的滑窗问题,但因为有了负数,滑窗思想不能用了。
通过分析,这道题应该属于我们上面列举四种情况的最后一种。具体思路如下:

  1. 初始化一个空的哈希表和total=0的前缀和变量
  2. 设置返回值res = 0,用于记录满足题意的子数组数量
  3. 循环数组的过程中,通过原地修改数组的方式,计算数组的累加和
  4. 将当前累加和减去整数K的结果,在哈希表中查找是否存在
  5. 如果存在该key值,证明以数组某一点为起点到当前位置满足题意,ret加等于将该key值对应的value
  6. 判断当前的累加和是否在哈希表中,若存在value+1,若不存在value=1
  7. 最终返回res即可

但在这里要注意刚才说到的前缀和边界问题。
我们在计算这种场景时,需要考虑如果以数组nums[0]为开头的连续子数组就满足题意呢?
此时候我们的哈希表还是空的,没办法计算前缀和!所以遇到这类题目,都需要在哈希表中默认插入一个{0:1}的键值对,
用于解决从数组开头的连续子数组满足题意的特殊场景。
下面就开始解题吧!

class Solution {
    public int subarraySum(int[] nums, int k) {
        int total = 0;
        int res = 0;
        Map<Integer,Integer> map = new HashMap<>();
        map.put(0,1);
        for (int i = 0; i < nums.length; i++) {
            total += nums[i];
            res += map.getOrDefault(total - k,0);
            map.put(total,map.getOrDefault(total,0)+1);
        }
        return res;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值