问题:
给定一个整数数组和一个整数 k,你需要找到该数组中和为 k 的连续的子数组的个数。
示例 :
输入:nums = [1,1,1],k = 2
输出: 2。[1,1] 与 [1,1] 为两种不同的情况。
说明 :
数组的长度为 [1, 20,000]。
数组中元素的范围是 [-1000, 1000] ,且整数 k 的范围是 [-1e7, 1e7]。
分析:
对于该问题,最简单直接的方法就是暴力法。将每个子数组的和与目标值 k 比较,得到结果。(暴力法就不过多介绍了)
代码如下:
public int subarraySum(int[] nums, int k) {
int length = nums.length, count = 0;
for(int i = 0; i < length; i++) {
int sum = 0;
for(int j = i; j < length; j++) {
sum += nums[j];
if(sum == k )
count++;
}
}
return count;
}
下面介绍一个新的题解:
我们假设有 [a, b] 是一个满足条件的区间,即有:数组的前 b 项和减去前 a 项和为目标值 k,sum(b) - sum(a) = k,即:前 b 项和减去目标值 k,得到前 a 项和,sum(b) - k = sum(a)。
这里就是重点了,我们用前 b 项和减去目标值 k,得到前 a 项和,若前 a 项和不存在,即找不到满足条件的区间 [a, b];若前 a 项和存在,则区间满足,将前 a 项和出现的次数,累加在 count 中。
我们需要一个 Map<Integer, Integer> sumAndCount,将每个区间和作为 key,区间和出现的次数作为 value,保存到 sumAndCount 中,若要保存的区间和已经存在,次数设置为 当前和出现次数 + 1,若不存在,次数设置为 1。
代码如下:
public int subarraySum(int[] nums, int k) {
int count = 0,sum = 0;
Map<Integer, Integer> sumAndCount = new HashMap<>();
// (0, 1) 是因为可能存在前 n 项和为 k 的情况,k - k = 0
sumAndCount.put(0,1);
for(int t : nums) {
// sum 保存第一项到当前项的和
sum += t;
// 判断是否存在某项和为 sum - k
// 若存在,将 sum - k 出现的次数累加到 count 中
if(sumAndCount.containsKey(sum-k))
count += sumAndCount.get(sum-k);
// 将当前和存到 sumAndCount 中
sumAndCount.put(sum, 1+sumAndCount.getOrDefault(sum,0));
}
return count;
}