560. 和为 K 的子数组

给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的连续子数组的个数 。

示例 1：

输入：nums = [1,1,1], k = 2
输出：2
示例 2：

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

公式

前i项和定义为preSum[i]
前j项和定义为preSum[j]
则preSum[j] - preSum[i] 为j-i项和
设preSum[j] - preSum[i] = k
则只需找到preSum[j] - k出现次数，即可知道n项和为k的次数
注意：maps[0] = 1 表示和为0出现过一次，考虑边界情况【0】 0。

思路

代码

class Solution {
    public int subarraySum(int[] nums, int k) {
        // 前缀和
        int count=0;
        Map<Integer,Integer> maps = new HashMap();
        int preSum = 0;

        // 0 和为0出现一次
        maps.put(0,1);
        for(Integer num:nums) {

            preSum += num;

            // 问题 转化 preSum[j] - preSum[i] = k 
            // k 为j-i求和
         
            count = count + maps.getOrDefault(preSum - k, 0);

            // 维护前缀和
            // 
            maps.put(preSum, maps.getOrDefault(preSum, 0) + 1);
        }

        return count;

    }
}