Create zero-filled-subarrays.py

gabedonnan · Mar 21, 2023 · f6fccaa · f6fccaa
1 parent cc5159c
commit f6fccaa
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/zero-filled-subarrays.py b/zero-filled-subarrays.py
@@ -0,0 +1,19 @@
+class Solution:
+    def zeroFilledSubarray(self, nums: List[int]) -> int:
+        flag = False
+        total = 0
+        for i in range(len(nums)):
+            if not flag and nums[i] == 0:
+                flag = True
+                startLoc = i
+            if flag and nums[i] != 0:
+                flag = False
+                total += self.addTotal(i - startLoc)
+        if flag:
+            total += self.addTotal(len(nums) - startLoc)
+        return total
+
+    @staticmethod
+    def addTotal(length: int) -> int:
+        length += 1
+        return int((length * (length-1)) / 2)