在由若干 0 和 1 组成的数组 A 中,有多少个和为 S 的非空子数组。
示例:
输入:A = [1,0,1,0,1], S = 2
输出:4
解释:
如下面黑体所示,有 4 个满足题目要求的子数组:
[1,0,1,0,1]
[1,0,1,0,1]
[1,0,1,0,1]
[1,0,1,0,1]
提示:
A.length <= 30000
0 <= S <= A.length
A[i] 为 0 或 1
思路:
这里只记录优秀的方法(借鉴)
使用一个变量 sum 来保存从第一个元素到目前为止的元素之和;
使用一个变量 result 来保存结果;
使用一个数组 arr,其索引为 i 的元素来保存从第一个元素开始和为 i 的子数组个数。(除了第一个元素,arr[0] = 1)
循环遍历数组 A,将当前的 num(A中的元素)加入 sum 总和中,如果 sum >= S,说明第一个元素到当前元素的和已经超过S,那么在 arr 中查找索引为(sum-S)的元素,加入到result里面。
/**
* @param {number[]} A
* @param {number} S
* @return {number}
*/
var numSubarraysWithSum = function(A, S) {
let result = 0;
let sum = 0;
let arr = Array(A.length + 1).fill(0)
arr[0] = 1
for (let num of A) {
sum += num;
if (sum >= S) {
result += arr[sum - S]
}
arr[sum] += 1
}
return result;
};