Solution for task 05.1 PassingCars

Author: moiseenkov

solution_05_1_passing_cars/__init__.py

@@ -0,0 +1,23 @@
+"""
+Codility exercise 05.1 PassingCars
+"""
+from solution_05_1_passing_cars.solution import solution
+
+
+def main():
+    """
+    Main function calls solution for codility task 05.1
+    """
+    while True:
+        try:
+            array = list(map(int, input('Enter int array with 0 or 1: ').strip().split(' ')))
+            print(solution(array))
+        except ValueError as ex:
+            print(f'Wring input! {ex}. Try again.')
+            continue
+        except KeyboardInterrupt:
+            return
+
+
+if __name__ == '__main__':
+    main()

solution_05_1_passing_cars/main.py

@@ -0,0 +1,18 @@
+"""
+Solution for exercise 05.1 PassingCars
+"""
+from itertools import accumulate
+
+
+def solution(a):
+    suffix_sum = list(accumulate(reversed(a)))
+    counter = 0
+    prev = suffix_sum[0]
+    for value in suffix_sum[1:]:
+        if value == prev:
+            counter += prev
+        else:
+            prev = value
+        if counter > 1000000000:
+            return -1
+    return counter