给你一个整数数组 nums 和一个整数 k ,请你统计并返回 该数组中和为 k 的子数组的个数 。
子数组是数组中元素的连续非空序列。
示例 1:
输入:nums = [1,1,1], k = 2
输出:2
示例 2:
输入:nums = [1,2,3], k = 3
输出:2
提示:
1 <= nums.length <= 2 * 104
-1000 <= nums[i] <= 1000
-107 <= k <= 107
1.暴力解法
class Solution {
public int subarraySum(int[] nums, int k) {
int sum = 0;
for (int i = 0; i < nums.length; i++) {
int sumi = nums[i];
if(sumi == k) {
sum++;
}
for (int j = i + 1; j < nums.length; j++) {
sumi += nums[j];
if(sumi == k) {
sum++;
}
}
}
return sum;
}
}
执行用时分布
1594ms
击败26.39%使用 Java 的用户
消耗内存分布
45.40MB
击败22.08%使用 Java 的用户
2.前缀和
class Solution {
public int subarraySum(int[] nums, int k) {
// 第一种方法的问题在于太多重复计算,
// 换一个思路,如果从第i个数到第j个数,这j-i+1个数组成的子数组之和为k
// 也就是数组nums的前j个数之和 减去 数组nums的前i个数之和 等于 k
// 所以可以先统计出来 前N个数之和,也就是 《前缀和》,这样就可以大幅度减少计算量
// 注意前面加了一个 0,这样便于处理第0位数的情况,看下面例子
// 对于 [3,1,2], k == 3 而言计算前缀和后转成了[0, 3, 4, 6]
// 就可以发现3-0==3, 6-3==3,就有两组子数组满足条件
int[] preSum = new int[nums.length + 1];
preSum[0] = 0;
for (int i = 0; i < nums.length; i++) {
preSum[i+1] = nums[i] + preSum[i];
}
// 还是硬扫,但采用从后往前遍历,即指定了上面提到的j,往前找一个i,使得preSum[j] - preSum[i] == k,其中j > i
int count = 0;
for (int j = 1; j < preSum.length; j++) {
int target = preSum[j] - k;
for (int i = j - 1; i >= 0; i--) {
if (preSum[i] == target) {
count++;
}
}
}
return count;
}
}
执行用时分布
1110ms
击败41.83%使用 Java 的用户
消耗内存分布
44.56MB
击败67.42%使用 Java 的用户