560. 和为 K 的子数组

560. 和为 K 的子数组

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

示例 1：

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

示例 2：

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

```

#### 思路

借助 `Map` 数据结构辅助计算，用变量 `pre` 记录当前遍历过的前缀和，用 `mp` 保存当前前缀和的数量（即出现次数）。初始化时，`mp.put(0,1)` 表示前缀和为 0 的出现次数为 1，因为在遍历过程中，当前遍历到的下标 i 后缀子数组的和也可能是 k，此时即使当前区间的和为 k，也要加上前缀和为 0 的情况。

当遍历到下标为 i 的位置时，更新当前前缀和 `pre`，然后计算当前前缀和减去 k 的值是否在 `mp` 中出现过，如果出现过，说明当前位置往前的某个位置到当前位置的子数组和等于 k，于是将出现次数加入计数器 `count` 中。此处采用 `mp.getOrDefault(pre - k, 0)` 方法来获取键值为 `pre - k` 的值，如果该键不存在于 `mp` 中，则默认值为 0。

最后，将出现次数加入 `mp` 中，方便后续的遍历继续计算。如果当前前缀和 `pre` 已经存在于 `mp` 中，则将其出现次数加 1。

#### 代码

class Solution {
    public int subarraySum(int[] nums, int k) {
        // 用 Map 记录前缀和 sum 出现的次数
        Map<Integer, Integer> mp = new HashMap<>();
        // 初始化，前缀和为 0 的出现次数为 1
        mp.put(0, 1);
        int count = 0;
        int pre = 0;
        for (int i = 0; i < nums.length; i++) {
            pre += nums[i];
            // 如果之前的某个位置 j 到 i 的子数组和为 k
            // 那么从 j+1 到 i 的子数组和就为 k
            if (mp.containsKey(pre - k)) {
                count += mp.get(pre - k);
            }
            // 将当前前缀和出现次数加入 mp 中
            mp.put(pre, mp.getOrDefault(pre, 0) + 1);
        }
        return count;
    }
}