求一个数组中连续数字的和为k的子数组个数。
一开始想到是滑动窗口,但是滑动窗口并不能解决,因为有负数,所以走过的还可能要用到。
又想到动态规划,也不行,因为状态转移式推不出来。
可以利用前缀和,前缀和就是用来解决连续子数组的和的问题,可以快速得到某个子数组的和,用一个前缀和数组来对原数组进行预处理。
若定义pre(i)是0到i的所有数的和,则一个j到i的子数组的和就可以用pre(j)和pre(i)来表示。
这个题的意思就是求多少个和为k的连续子数组,则最直接的想法是就看以i为结尾的数组中,有多少个是满足他们的前缀和是等于k的。但这样是O(n2)
所以如果想要改进,改一下表达式pre(j)=pre(i)-k时成立
可以用一个map来在存前缀和的同时,存下该前缀和出现的次数,所以在遍历到i的时候要找有几个pre(i)-k的前缀和时可以直接得到次数,就不用for循环遍历找了。
public class Solution {
public int subarraySum(int[] nums, int k) {
int count = 0, pre = 0;
HashMap <Integer, Integer> map = new HashMap <Integer, Integer>();//存的是此时的pre值对应的次数
map.put(0, 1);//当pre为0时次数为1
for (int i = 0; i < nums.length; i++) {//求每个i有几个pre[i]-k
pre += nums[i];//算此时i的pre数组
if (map.containsKey(pre - k)) {//如果拥有则加到次数中
count+=map.get(pre - k);
}
map.put(pre, map.getOrDefault(pre, 0) + 1);//把此时的pre次数更新到map中
}
return count;
}
}