930 accepted.

Author: aQuaYi

Algorithms/0930.binary-subarrays-with-sum/binary-subarrays-with-sum.go

@@ -1,6 +1,10 @@
 package problem0930
 
 func numSubarraysWithSum(A []int, S int) int {
+	if S == 0 {
+		return zero(A)
+	}
+
 	size := len(A)
 	i, j, sum := 0, 0, 0
 
@@ -10,9 +14,9 @@ func numSubarraysWithSum(A []int, S int) int {
 	}
 
 	res := 0
-	for i < size && j <= size {
+	for i < size && j <= size && sum == S {
 		left := 1
-		for i < size && A[i] == 0 {
+		for i < j && A[i] == 0 {
 			i++
 			left++
 		}
@@ -31,3 +35,17 @@ func numSubarraysWithSum(A []int, S int) int {
 
 	return res
 }
+
+func zero(A []int) int {
+	res := 0
+	tmp := 0
+	for _, a := range A {
+		if a == 1 {
+			tmp = 0
+		} else {
+			tmp++
+			res += tmp
+		}
+	}
+	return res
+}

Algorithms/0930.binary-subarrays-with-sum/binary-subarrays-with-sum_test.go

@@ -13,6 +13,18 @@ var tcs = []struct {
 	ans int
 }{
 
+	{
+		[]int{0, 0, 0, 0, 0, 0, 1, 0, 0, 0},
+		0,
+		27,
+	},
+
+	{
+		[]int{0, 0, 0, 0, 1},
+		2,
+		0,
+	},
+
 	{
 		[]int{0, 0, 0, 0, 0},
 		0,