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

一、题目

给你一个整数数组 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;
    }
}

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