【LeetCode 热题 100】和为 K 的子数组（前缀和+哈希）

问题描述

给你一个整数数组 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

AC代码

class Solution {
    public int subarraySum(int[] nums, int k) {
        int count = 0, pre = 0;     // count 子数组个数，pre为前缀和
        Map<Integer, Integer> map = new HashMap<>();    // 键为前缀和，值为该前缀和出现的次数
        map.put(0, 1);  // 考虑前缀和为0的情况
        for (int num : nums) {      // 遍历数组
            pre += num;     // 维护前缀和
            count += map.getOrDefault(pre - k, 0);  // 累加前缀和为pre-k的个数，即符合题意的子数组个数
            map.put(pre, map.getOrDefault(pre, 0) + 1);  // 更新前缀和为pre的个数
        }
        return count;
    }
}

（by 归忆）