diff options
Diffstat (limited to 'scratch/facebook/merging-ranges.py')
-rw-r--r-- | scratch/facebook/merging-ranges.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/scratch/facebook/merging-ranges.py b/scratch/facebook/merging-ranges.py new file mode 100644 index 000000000000..6da44572ee7e --- /dev/null +++ b/scratch/facebook/merging-ranges.py @@ -0,0 +1,23 @@ + +def merge(xs): + xs.sort() + result = xs[0:1] + for a, b in xs[1:]: + y, z = result[-1] + if a <= z: + result[-1] = (y, max(b, z)) + else: + result.append((a, b)) + return result + +inputs = [([(0,1),(3,5),(4,8),(10,12),(9,10)], [(0,1),(3,8),(9,12)]), + ([(1,2),(2,3)], [(1,3)]), + ([(1,5),(2,3)], [(1,5)]), + ([(1,10),(2,6),(3,5),(7,9)], [(1,10)]), + ] +for x, expected in inputs: + result = merge(x) + print(x) + print(result) + assert result == expected + print("Success!") |