题目描述:
给你一个整数数组 nums
和一个整数 k
,请你统计并返回 该数组中和为 k
的子数组的个数。子数组是数组中元素的连续非空序列。
示例 1:
输入:nums = [1,1,1], k = 2
输出:2
示例 2:
输入:nums = [1,2,3], k = 3
输出:2
题解:
前缀和+哈希表
class Solution {
public int subarraySum(int[] nums, int k) {
int n = nums.length;
int[] s = new int[n + 1];
for (int i = 0; i < n; i++) {
s[i + 1] = s[i] + nums[i];
}
int ans = 0;
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i <= n; i++) {
ans += map.getOrDefault(s[i] - k, 0);
map.merge(s[i], 1, Integer :: sum);
}
return ans;
}
}