Solution for task 04.4 MissingInteger

Author: moiseenkov

solution_04_4_missing_integer/__init__.py

@@ -0,0 +1,23 @@
+"""
+Codility exerciece 04.4 MissingInteger
+"""
+from solution_04_4_missing_integer.solution import solution
+
+
+def main():
+    """
+    Main function calls solution for codility task 04.4
+    """
+    while True:
+        try:
+            array = list(map(int, input('Enter int array: ').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_04_4_missing_integer/main.py

@@ -0,0 +1,16 @@
+"""
+Solution for exercise 04.4 MissingInteger
+"""
+
+
+def solution(a):
+    max_elem = max(a)
+    if max_elem < 1:
+        return 1
+    source_set = set(a)
+    complete_set = set(range(1, max_elem + 1))
+    diff = complete_set - source_set
+    if diff:
+        return min(diff)
+    else:
+        return max_elem + 1