这道题的暴力解法比较容易想到,即固定一边界去遍历另一边界,统计所有的子数组的和为k的数量,时间复杂度O(N^2)
基于HashMap的实现算法
1.前缀和sum[N]:数组中前N项的和,nums[0]+…+nums[N-1]。
2.sum[N]-sum[M]=sum[M]+…sum[N-1],通过这种方式可以统计所有的子数组个数。
3.这道题目仅要求求出个数,所以无需关注子数组位置,仅统计数量即可。
HashMap中存储的是前缀和,并且将前缀和作为键,前缀和出现的次数作为值。
上图为例:数组为[3, 4, 5, 3, -3]
1.首先将(0, 1)放入HashMap,即将和为0的数组个数置为1,这一步是必须的,下面会证明
2.不断地统计前缀和,下一步该统计的是0位置的3,sum=3。此时计算HashMap中是否包含(sum[0]-k=0)的键,发现包含,且个数为1,这一步的含义是,统计数组到sum计算的位置0为止,包含的所有子数组中和为k的个数,将(0, 1)放进HashMap中的作用是要包含前缀和为k的情况,即从0开始加到N项正好为k的情况,如果不添加(0,1)是没有统计这种情况的,可以拿上面的例子实验一下。
3.接下来就是从0到数组结尾遍历,统计子数组和为k的情况。
public int subarraySum(int[] nums, int k) {
// count统计数量
int count=0, sum=0;
// 统计前缀和次数
HashMap<Integer,Integer> map=new HashMap<>();
// 这一步的作用是使后面的map[k]-k=map[0]存在1个,但是此0自身不算在计数内
map.put(0,1);
for (int i=0;i<nums.length;i++){
sum += nums[i];
if (map.containsKey(sum-k)){
// 添加对应的value
count += map.get(sum-k);
}
// key中有sum值则value+1,无sum值则为1
map.put(sum, map.getOrDefault(sum,0)+1);
}
return count;
}
语言表达能力有限,有什么问题可以评论区提问。