LeetCode——930. 和相同的二元子数组（JavaScript）

在由若干 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;
};