From e98fa50d9dd47dd02ce01b6db838fd05ec0e1a97 Mon Sep 17 00:00:00 2001 From: Lubos Kolouch Date: Tue, 14 Feb 2023 11:00:50 +0100 Subject: Challenge 201 Task1 LK Python --- challenge-201/lubos-kolouch/python/ch-1.py | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) create mode 100644 challenge-201/lubos-kolouch/python/ch-1.py diff --git a/challenge-201/lubos-kolouch/python/ch-1.py b/challenge-201/lubos-kolouch/python/ch-1.py new file mode 100644 index 0000000000..dc7229e912 --- /dev/null +++ b/challenge-201/lubos-kolouch/python/ch-1.py @@ -0,0 +1,28 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +import unittest + + +def find_missing_numbers(arr): + n = len(arr) + s = set(arr) + missing_numbers = [] + for i in range(n + 1): + if i not in s: + missing_numbers.append(i) + return missing_numbers + + +# test cases +class TestMissingNumbers(unittest.TestCase): + + def test_example1(self): + self.assertEqual(find_missing_numbers([0, 1, 3]), [2]) + + def test_example2(self): + self.assertEqual(find_missing_numbers([0, 1]), [2]) + + +if __name__ == '__main__': + unittest.main() -- cgit