和为K的子数组

和为K的子数组

题目：
给定一个整数数组和一个整数 k，你需要找到该数组中和为 k 的连续的子数组的个数。

示例 1 :

输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。

解题思路：计算数组的前缀和，并用hashmap来记录当前前缀和与k的差值的出现次数即可

class Solution {
    public int subarraySum(int[] nums, int k) {
        int sum = 0, len = nums.length, ans = 0;
        Map<Integer, Integer> map = new HashMap();
        map.put(0, 1);
        for(int i = 0; i < len; i++) {
            sum += nums[i];
            if(map.containsKey(sum - k)) {
                ans += map.get(sum - k);
            }
            
            map.put(sum, map.getOrDefault(sum, 0) + 1);
        }
        
        return ans;
    }
}