算法小解：560. 和为 K 的子数组

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

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

提示：

• 1 <= nums.length <= 2 * 10^4
• -1000 <= nums[i] <= 1000
• -10^7 <= k <= 10^7

解法：使用前缀和以及哈希表，使用一个数组s记录前缀和，寻找和为k的子数组只需要知道s[i] = s[j] - k（i < j）是否存在值为s[i]的前缀和，如果存在就查看有多少个这样的前缀和（用哈希表记录）即可。

hashmap.merge(key, value, remappingFunction)：如果 key 对应的 value 不存在，则返回该 value 值，如果存在，则返回通过 remappingFunction 重新计算后的值。

思路：灵茶山艾府

class Solution {
    public int subarraySum(int[] nums, int k) {
        int n = nums.length;
        int[] s = new int[n + 1]; // 前缀和
        int ans = 0;
        Map<Integer, Integer> cnt = new HashMap<>(n + 1); // 前缀和计数器
        cnt.merge(0, 1, Integer::sum); // 先将 s[0] 记录进来
        for(int i = 0; i < n; i++){
            s[i + 1] = s[i] + nums[i];
            ans += cnt.getOrDefault(s[i + 1] - k, 0);
            cnt.merge(s[i + 1], 1, Integer::sum);
        }
        return ans;
    }
}

优化：可以s只需要使用一次就可以了，所以不需要使用数组记录。

class Solution {
    public int subarraySum(int[] nums, int k) {
        int n = nums.length;
        int ans = 0;
        int s = 0;
        Map<Integer, Integer> cnt = new HashMap<>(n + 1); // 前缀和计数器
        cnt.merge(0, 1, Integer::sum); // 先将 s[0] 记录进来
        for (int i = 0; i < n; i++) {
            s += nums[i];
            ans += cnt.getOrDefault(s - k, 0);
            cnt.merge(s, 1, Integer::sum);
        }
        return ans;
    }
}