Medium
一刷的笔记是直接复制粘贴过来的别人的语言,果然就完全没印象。虽然看似快,其实这样是最事倍功半的。一定要懂了,用自己的语言简单总结一下就好。别直接copy,没有意思。
第一种解法是O(N2)的brute force,就不写了。第二种方法是O(N)利用sum[i, j] = sum[0, j] - sum[0, i-1]
where sum[i,j] stands for sum for substring[i,j] inclusively. 利用preSum这个装着preSum和它的频次的map, 遍历是一边加sum以便存入preSum.一般检查有没有之前的presum = sum – k的,为什么是sum – k的呢?因为preSum[0,i] + subarraySum = preSum[0, j], 知道 subarraySum = k, preSum[0, j] = sum, 所以要找有没有preSum[0,i] = sum – k的。能找到几个这样的preSum, 就能找到几个与之对应的subarray, 所以res++.
class Solution {
public int subarraySum(int[] nums, int k) {
if (nums == null || nums.length == 0){
return 0;
}
int res = 0;
int sum = 0;
Map<Integer, Integer> preSum = new HashMap<>();
preSum.put(0, 1);
for (int i = 0; i < nums.length; i++){
sum += nums[i];
if (preSum.containsKey(sum - k)){
res += preSum.get(sum - k);
}
preSum.put(sum, preSum.getOrDefault(sum, 0) + 1);
}
return res;
}
}