🔄Update: 和为K的子数组
This commit is contained in:
@@ -0,0 +1,59 @@
|
||||
# 和为 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;
|
||||
}
|
||||
}
|
||||
```
|
||||
Reference in New Issue
Block a user