Hot100【二十二】：和为 K 的子数组

// 用前缀和和hashmap来做
// 这个算法思想和常规的不一样： 比如nums = [1,2,3], k = 3, ，比如map里面存0，1，3，6，用1，3，6去减3，如果发现map中存在，就说明是存在和为k的这一段的
class Solution {
    public int subarraySum(int[] nums, int k) {
        // 1. 不需要特殊处理

        int prefixSum = 0; // 前缀和
        int count = 0; // 一个有多少个
        // 2. 创建一个map
        Map<Integer, Integer> prefixSum2Count = new HashMap<>();
        // 3. 往里面放一个和为0的一段
        prefixSum2Count.put(0, 1);

        for (int i = 0; i < nums.length; i++) {
            prefixSum += nums[i];
            // 4. 看看前缀和-k 存在不，也就是说之前有没有一段和为k的前缀
            if (prefixSum2Count.get(prefixSum - k) != null) {
                count += prefixSum2Count.get(prefixSum - k);
            }
            // 5. 把当前前缀和放到map中
            prefixSum2Count.put(prefixSum, prefixSum2Count.getOrDefault(prefixSum, 0) + 1);
        }
        return count;
    }

}