力扣560和为K的子数组

560. 和为 K 的子数组

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

示例 1：

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

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

提示：

1 <= nums.length <= 2 * 10⁴
-1000 <= nums[i] <= 1000
-10⁷ <= k <= 10⁷

思路：前缀和+哈希表。一个连续子区间[j,i]的和==k相当于是前缀和[i] - 前缀和[j]==k，所以进一步把问题转换成前缀和[i] - k == 前缀和[j]，也就是去找有多少个前缀和[j]在前缀和i之前，这时候又可以用哈希表来记录前缀和[j]的数量，要注意的是特殊情况，当第一个nums[0]就等于k，后面就无法取map，所以令map有个默认值{0,1}

class Solution {
    public int subarraySum(int[] nums, int k) {
        int[] sum = new int[nums.length];
        sum[0] = nums[0];
        //计算前缀和
        for(int i = 1; i < nums.length; i++){
            sum[i] = sum[i - 1] + nums[i];
        }
        Map<Integer, Integer> map = new HashMap<>();
        map.put(0, 1); //特殊情况，nums[0]就满足k
        int res = 0;
        for(int i = 0; i < nums.length; i++){
            int t = sum[i] - k;
            res += map.getOrDefault(t, 0);
            map.put(sum[i],map.getOrDefault(sum[i], 0) + 1); //统计有多少个和sum[i]相同的sum[j]
        }
        return res;
    }
}