剑指 Offer II 010. 和为 k 的子数组
题目
给定一个整数数组和一个整数k ,请找到该数组中和为k的连续子数组的个数。
示例 1 :
输入:nums = [1,1,1], k = 2
输出: 2
解释: 此题 [1,1] 与 [1,1] 为两种不同的情况
示例 2:
输入:nums = [1,2,3], k = 3
输出: 2
提示:
- 1 <= nums.length <= 2 * 104
- -1000 <= nums[i] <= 1000
- -107 <= k <= 107
注意:本题与主站560题相同
题解
public int subarraySum(int[] nums, int k) {
int result = 0, sum = 0;
Map<Integer, Integer> prefix = new HashMap<>();
// 前缀和为0的个数 仅作为初始化
prefix.put(0, 1);
for (int num : nums) {
sum += num;
// 若存在前缀和为sum-k则为解
result += prefix.getOrDefault(sum - k, 0);
// 合并前缀和为sum的子数组数量
prefix.merge(sum, 1, Integer::sum);
}
return result;
}