1018. Binary Prefix Divisible By 5

donutloop · donutloop · commit 32727a1c9d96 · 2024-04-03T20:09:12.000+02:00
diff --git a/Binary Prefix Divisible By 5/kata.py b/Binary Prefix Divisible By 5/kata.py
@@ -0,0 +1,30 @@
+class Solution(object):
+    def prefixesDivBy5(self, nums):
+        """
+        :type nums: List[int]
+        :rtype: List[bool]
+        """
+        result = [False] * len(nums)  
+        tracker = 1  
+        i = 0
+        for i, num in enumerate(nums):
+            if num != 0:  
+                break
+            else:
+                result[i] = True  
+
+        i += 1  
+        for num in nums[i:]:
+            if num == 0:
+                tracker = tracker << 1  
+                if tracker % 5 == 0:
+                    result[i] = True
+            else:
+                tracker = tracker << 1  
+                tracker = tracker | 1  
+                if tracker % 5 == 0:
+                    result[i] = True
+            i += 1
+
+        return result
+        
