59 lines
1018 B
Markdown
59 lines
1018 B
Markdown
|
# 和为 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
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
## 思路
|
|||
|
|
|
|||
|
|
- 前缀和
|
|||
|
|
- `prefixSum`
|
|||
|
|
- 哈希表单次遍历
|
|||
|
|
|
|||
|
|
## 代码
|
|||
|
|
|
|||
|
|
```java
|
|||
|
|
class Solution {
|
|||
|
|
public int subarraySum(int[] nums, int k) {
|
|||
|
|
// Init
|
|||
|
|
// Prefix Sum
|
|||
|
|
Map<Integer, Integer> map = new HashMap<>();
|
|||
|
|
map.put(0, 1);
|
|||
|
|
int ans = 0, pre = 0;
|
|||
|
|
|
|||
|
|
// Traverse
|
|||
|
|
for (int i = 0; i < nums.length; i++) {
|
|||
|
|
pre += nums[i];
|
|||
|
|
// Judge
|
|||
|
|
if (map.containsKey(pre - k)) {
|
|||
|
|
ans += map.get(pre - k);
|
|||
|
|
}
|
|||
|
|
map.put(pre, map.getOrDefault(pre, 0) + 1);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return ans;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
```
|