import unittest import timeit # Solution that uses O(n) space to store the result. def not_in_place(xs): xs.sort() result = [xs[0]] for ca, cb in xs[1:]: pa, pb = result[-1] if ca <= pb: result[-1] = (pa, max(pb, cb)) else: result.append((ca, cb)) return result # Solution that uses O(1) space to store the result. def in_place(xs): xs.sort() i = 0 j = i + 1 while j < len(xs): pa, pb = xs[i] ca, cb = xs[j] if ca <= pb: xs[i] = (pa, max(pb, cb)) del xs[j] else: i = j j += 1 return xs def test_nip(): inputs = [ [(1, 3), (2, 4)], [(5, 6), (6, 8)], [(1, 8), (2, 5)], [(1, 3), (4, 8)], [(1, 4), (2, 5), (5, 8)], [(5, 8), (1, 4), (6, 8)], [(1, 10), (2, 5), (6, 8), (9, 10), (10, 12)], [(0, 1), (3, 5), (4, 8), (10, 12), (9, 10)], ] for x in inputs: not_in_place(x) def test_ip(): inputs = [ [(1, 3), (2, 4)], [(5, 6), (6, 8)], [(1, 8), (2, 5)], [(1, 3), (4, 8)], [(1, 4), (2, 5), (5, 8)], [(5, 8), (1, 4), (6, 8)], [(1, 10), (2, 5), (6, 8), (9, 10), (10, 12)], [(0, 1), (3, 5), (4, 8), (10, 12), (9, 10)], ] for x in inputs: in_place(x) merge_ranges = in_place setup = 'from __main__ import test_nip, test_ip' print(timeit.timeit('test_nip()', number=10000, setup=setup)) print(timeit.timeit('test_ip()', number=10000, setup=setup)) # Tests class Test(unittest.TestCase): def test_meetings_overlap(self): actual = merge_ranges([(1, 3), (2, 4)]) expected = [(1, 4)] self.assertEqual(actual, expected) def test_meetings_touch(self): actual = merge_ranges([(5, 6), (6, 8)]) expected = [(5, 8)] self.assertEqual(actual, expected) def test_meeting_contains_other_meeting(self): actual = merge_ranges([(1, 8), (2, 5)]) expected = [(1, 8)] self.assertEqual(actual, expected) def test_meetings_stay_separate(self): actual = merge_ranges([(1, 3), (4, 8)]) expected = [(1, 3), (4, 8)] self.assertEqual(actual, expected) def test_multiple_merged_meetings(self): actual = merge_ranges([(1, 4), (2, 5), (5, 8)]) expected = [(1, 8)] self.assertEqual(actual, expected) def test_meetings_not_sorted(self): actual = merge_ranges([(5, 8), (1, 4), (6, 8)]) expected = [(1, 4), (5, 8)] self.assertEqual(actual, expected) def test_one_long_meeting_contains_smaller_meetings(self): actual = merge_ranges([(1, 10), (2, 5), (6, 8), (9, 10), (10, 12)]) expected = [(1, 12)] self.assertEqual(actual, expected) def test_sample_input(self): actual = merge_ranges([(0, 1), (3, 5), (4, 8), (10, 12), (9, 10)]) expected = [(0, 1), (3, 8), (9, 12)] self.assertEqual(actual, expected) unittest.main(verbosity=2)