leetcode 560
560. 和为K的子数组
给定一个整数数组和一个整数 k,你需要找到该数组中和为 k 的连续的子数组的个数。
例:输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。
利用前缀和:求某个区间的累加和
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];
}
preSum[i]就是nums[0…i-1]的和。
求 nums[i…j]的和,那么有preSum[j+1] - preSum[i]
public int subarraySum(int[] nums, int k) {
int len=nums.length;
int[] sum = new int[len+1];
sum[0]=0;
for(int i=0;i<len;i++){
sum[i+1]=sum[i]+nums[i];
}
int ans =0;
for(int i=1;i<=len;i++){
for(int j=0;j<i;j++){
if(sum[i]-sum[j]==k)
ans++;
}
}
return ans;
}