题目来源:题目
题目
给定一个整数数组和一个整数 k,你需要找到该数组中和为 k 的连续的子数组的个数。
示例 1 :
输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。
说明 :
数组的长度为 [1, 20,000]。
数组中元素的范围是 [-1000, 1000] ,且整数 k 的范围是 [-1e7, 1e7]。
思路
思路一(枚举)
计算每个以下标j结尾的子数组和,判断该子数组的和是否为k。下标为j的子数组有j+1个,但是计算子数组的和时从后往前计算就比较简单,比如知道了子数组[i,j]的和,那么[i-1,j]的和就是[i,j]的和加上nums[i-1]。
思路二(前缀和+哈希表)
我们把下标为从0-j的数组和叫做j个前缀和。记作pre[j]。
假如下标从i-j的子数组和为k。就有pre[j] – pre[i – 1] == k.
即pre[i-1] == pre[j] – k 这个公式中有0 <= i <= j
于是我们可以计算每个pre[j] 的值,然后判断之前计算的前缀和中pre[j] – k的值有多少个,就是和为k的子数组个数。
可以用哈希表的键来存储pre[j],用哈希表的键值对应前缀和为pre[j]的个数。这样在查询的时候时间复杂度就为O(1)。
代码
思路一代码
public static int subarraySum(int[] nums, int k) {
int result = 0;
for(int end = 0;end < nums.length;end++){
int sum = 0;
for(int start = end;start >= 0;start--){
sum += nums[start];
if(sum == k)result++;
}
}
return result;
}
思路二代码
public static int subarraySum(int[] nums, int k) {
int pre = 0,result = 0;
Map<Integer,Integer> mp = new HashMap<>();
mp.put(0,1);
for(int i = 0;i < nums.length;i++){
pre += nums[i];
if(mp.containsKey(pre - k))result += mp.get(pre - k);
mp.put(pre,mp.getOrDefault(pre,0)+1);
}
return result;
}
总结
思路一的时间复杂度为O(n^2),空间复杂度为O(1)
思路二的时间复杂度为O(n),空间复杂度为O(n)