560. 和为 K 的子数组
560. 和为 K 的子数组
拿到题,一看到数组中和
为k,我就想到了用前缀和来做。定义一个前缀和函数perSum
,长度为num+1
。perSum[i],表示sum[0]
到sum[i-1]
这些值的和。那么很容易就知道sum[i]
到sum[j]
这个区间的和为perSum[j+1]-perSum[i]
,直接看代码。
class Solution {
public int subarraySum(int[] nums, int k) {
//用来返回值
int res = 0;
//边界值
if(nums.length == 0){
return res;
}
int[] perSum = new int[nums.length+1];
// 构造前缀和
for(int i = 1; i< perSum.length; i++){
perSum[i] = perSum[i-1] + nums[i-1];
}
for(int i=0 ; i < perSum.length; ++i){
for(int j=i+1 ; j < perSum.length ; ++j){
if(perSum[j]-perSum[i]==k){
res++;
}
}
}
return res;
}
}
前缀和也是一种做题技巧。