给你一个整数数组 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
前缀和preSum一般使用需要求解在一维数组或者多维数组的某个区间段所有元素之和,前提条件是原数组保持不变
解法一:时间复杂度O(n^2)
class Solution {
public int subarraySum(int[] nums, int k) {
int n = nums.length;
//preSum[i]记录nums[0,……i-1]总和
int[] preSum = new int[n+1];
preSum[0] = 0;
for(int i=1;i<=n;i++){
preSum[i] = preSum[i-1] + nums[i-1];
}
int count = 0;
for(int i=1;i<preSum.length;i++){
for(int j=0;j<i;j++){
//nums[j……i-1]之和
if(preSum[i]-preSum[j]==k)
count++;
}
}
return count;
}
}
解法二:利用HashMap减少内层循环,记录前缀和的同时也记录该前缀和出现的次数
class Solution {
public int subarraySum(int[] nums, int k) {
int n = nums.length;
//前缀和->前缀和出现次数
HashMap<Integer,Integer> preSum = new HashMap<>();
preSum.put(0,1);
int count = 0;
//用于记录每次进入循环后的前缀和
int sum_i = 0;
for(int i=0;i<n;i++){
sum_i += nums[i];
//符合该次循环的前缀和,即sum_i+sum_j=k
int sum_j = sum_i - k;
//如果前面存在sum_j这个符合条件的前缀和,则直接更新count
if(preSum.containsKey(sum_j))
count += preSum.get(sum_j);
//将新的前缀和sum_i加入HashMap并记录出现次数
preSum.put(sum_i,preSum.getOrDefault(sum_i,0)+1);
}
return count;
}
}