- 前缀和例题,欢迎大家来尝试。
给你一个整数数组 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
public static int subarraySum(int[] nums, int k) {
int len = nums.length;
// 计算前缀和数组
int[] preSum = new int[len + 1];
preSum[0] = 0;
for (int i = 0; i < len; i++) {
preSum[i + 1] = preSum[i] + nums[i];
}
int count = 0;
//暴力计算前缀和结果s[right+1]-s[left] == k的值
for (int left = 0; left < len; left++) {
//注意这里right的下标要等于left,否则会导致重复计算
for (int right = left; right < len; right++) {
if (preSum[right + 1] - preSum[left] == k){
count++;
}
}
}
return count;
}