560. 和为 K 的子数组（java）

题目：

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

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

示例 1：

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

示例 2：

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

提示：

• 1 <= nums.length <= 2 * 104
• -1000 <= nums[i] <= 1000
• -107 <= k <= 107

思路：其实这道题可以使用前缀和+哈希的做法来实现的。

代码：

class Solution {
    public int subarraySum(int[] nums, int k) {
        long[] prefix = new long[nums.length]; // 前缀和数组
        Map<Long, Long> hash = new HashMap<>();
        int temp = 0;
        int res = 0;
        for (int i = 0; i < nums.length; i++) {
            temp += nums[i];
            prefix[i] = temp;
        }
        for (int i = 0; i < nums.length; i++) {
            if (prefix[i] == k) res ++;
            if (hash.containsKey(prefix[i] - k)) res += hash.get(prefix[i] - k);
            // 更新 hash 数组
            if (hash.containsKey(prefix[i])) hash.put(prefix[i], hash.get(prefix[i]) + 1);
            else hash.put(prefix[i], 1L);
        }
        return res;
    }
}