「优选算法刷题」:和为K的子数组

67 阅读1分钟

一、题目

给你一个整数数组 nums 和一个整数 k ,请你统计并返回

该数组中和为 k 的子数组的个数

子数组是数组中元素的连续非空序列。

示例 1:

输入:nums = [1,1,1], k = 2
输出:2

示例 2:

输入:nums = [1,2,3], k = 3
输出:2

提示:

  • 1 <= nums.length <= 2 * 104
  • -1000 <= nums[i] <= 1000
  • -107 <= k <= 107

二、思路解析

这道题我用的是 hash 表加上前缀和的做法,通过 hash 表记录 nums 数组的前缀和,然后数一数 hash 表中值为 sum - k 的元素个数即可。

这其中有边界情况需要考虑:当数组起始位置到当前位置的元素和正好等于 k,那么根据公式 sum - k 就会得到 0,但是 hash 表中此时还没有计数呢,所以我们还需要手动插入一个 (0 , 1) 到 hash 表,以避免这种边界情况的发生。

然后就是更新返回值和 hash 表的操作了,具体解析我放在了下面的代码注释中👇

三、完整代码

class Solution {
    public int subarraySum(int[] nums, int k) {
        Map<Integer,Integer> hash = new HashMap<Integer,Integer>();
        hash.put(0,1);
        int sum = 0;
        int ret = 0;
        for(int x : nums){
            sum += x;
            ret += hash.getOrDefault(sum - k , 0); 
            // 查找 hash 中是否存在一个键值对,其键为 sum - k。
            // 如果存在,则取出对应的值并累加到 ret 上。
            // 检查是否存在之前的某个累加和,使得从那个位置到当前位置的子数组和正好为 k。

            hash.put(sum,hash.getOrDefault(sum,0) + 1);
            // 更新 hash 映射,将当前的累加和 sum 作为键,其出现的次数加 1 作为值。    
            // 如果 sum 未在 hash 中出现过,使用默认值 0,并加 1 后存入。

        }
        return ret;
    }
}

以上就是本篇博客的全部内容啦,如有不足之处,还请各位指出,期待能和各位一起进步!