题目
给定一个整数数组和一个整数 k ,请找到该数组中和为 k 的连续子数组的个数。
代码实现
因为有负数,所以我们不能使用滑动窗口
前缀和+双重遍历
效率很低
class Solution {
public int subarraySum(int[] nums, int k) {
int total = 0;
int res = 0;
for (int i = 1; i < nums.length; i++) {
nums[i] = nums[i-1] + nums[i];
}
for (int i = 0; i < nums.length; i++) {
if(nums[i] == k){
res++;
}
for (int j = 0; j < i; j++) {
if(nums[i] - nums[j] == k){
res++;
}
}
}
return res;
}
}
前缀和+Map
大佬思路:
这道题目非常简洁,就是求数组中何为整数k的连续子数组个数。
如果这道题的取值没有负数,那就是标准的滑窗问题,但因为有了负数,滑窗思想不能用了。
通过分析,这道题应该属于我们上面列举四种情况的最后一种。具体思路如下:
- 初始化一个空的哈希表和total=0的前缀和变量
- 设置返回值res = 0,用于记录满足题意的子数组数量
- 循环数组的过程中,通过原地修改数组的方式,计算数组的累加和
- 将当前累加和减去整数K的结果,在哈希表中查找是否存在
- 如果存在该key值,证明以数组某一点为起点到当前位置满足题意,ret加等于将该key值对应的value
- 判断当前的累加和是否在哈希表中,若存在value+1,若不存在value=1
- 最终返回res即可
但在这里要注意刚才说到的前缀和边界问题。
我们在计算这种场景时,需要考虑如果以数组nums[0]为开头的连续子数组就满足题意呢?
此时候我们的哈希表还是空的,没办法计算前缀和!所以遇到这类题目,都需要在哈希表中默认插入一个{0:1}的键值对,
用于解决从数组开头的连续子数组满足题意的特殊场景。
下面就开始解题吧!
class Solution {
public int subarraySum(int[] nums, int k) {
int total = 0;
int res = 0;
Map<Integer,Integer> map = new HashMap<>();
map.put(0,1);
for (int i = 0; i < nums.length; i++) {
total += nums[i];
res += map.getOrDefault(total - k,0);
map.put(total,map.getOrDefault(total,0)+1);
}
return res;
}
}