Haskell Skyline
Recently I started learning Haskell by studying the Intro to FP Programming course on Edx. Since then, I try to model different problems using Haskell.
One of these problems is the Skyline problem, which goes like this:
You are given a set of rectangular buildings in a city, and you should return the skyline view of the city. Input is a sequence of tuples
(x_{left}, height, x_{right})
, each describing a building. The output is a sequence of pairs(x, height)
meaning that the height of skyline changed toheight
at the given x coordinate.
For example, for the input:
[(1,11,5), (2,6,7), (3,13,9), (12,7,16), (14,3,25), (19,18,22),
(23,13,29), (24,4,28)]
The output should be:
[(1,11),(3,13),(9,0),(12,7),(16,3),(19,18),(22,3),(23,13),(29,0)]
Below diagrams show the input buildings and the expected skyline side by side:
Solving the problem
You can find an extended analysis of this problem at Brian Gordon’s blog , which explains several solutions of $O(n^2)$ and O(n \log n)
complexity to this problem.
(See comments)
In this post I’m not concerned with the time complexity at all, but mostly with the simplicity of the solution.
It’s also okay if a solution if contains a false height change, i.e. if we have two consecutive items in the output that contain the same height. After all, we are only concerned with how the skyline will look, and having a false change doesn’t change the shape of skyline.
Solution 1. Calculate height of skyline at all endpoints
We’re only interested at changes to the height of skyline, and height can only change where a building starts or ends. The height at a point can be calculated by finding the max height of buildings overlapping that point.
This leads us to the following solution:
1 2 3 4 5 6 |
skyline bs = sort (foldl add_endpoints [] bs) where add_endpoints xs (x1, h, x2) = xs ++ [(x1, height x1), (x2, height x2)] height x = maximum (0 : [h| (x1, h, x2) <- bs, x >= x1 && x < x2]) |
This solution has time complexity of O(n^2)
and produces some false changes, but it is very simple and easy to understand.
(We can easily remove the false changes by passing the result through another function, but this is good enough for our purpose and our goal is to keep things simple).
Solution 2. Iteratively add buildings
One might think we can iteratively add buildings and update the shape of skyline. Since a building can be taller or shorter than the previously added buildings, we may need to handle several cases to make this work correctly.
But if we sort the buildings by height before adding them, we can make the update much simpler. If we know the height of skyline is shorter than current building, then all we need to do is to remove the previous points between the boundries of the current building (since they won’t be visible anymore), and then add the two points of building.
This leads us to the following solution:
1 2 3 4 5 6 7 |
skyline bs = sort (foldl add_building [] (sortWith (\(_,h,_)->h) bs)) where height xs y = (snd . maximum) ((0, 0): [(x, h)| (x, h) <- xs, x <= y]) add_building xs (x1, h, x2) = [(x, h)| (x, h) <- xs, x < x1 || x > x2] ++ [(x1, h), (x2, height xs x2)] |
Update. Initial version of this code, which wrongly added (x2, 0)
in add_building
was incorrect, for the reason described by Chris_Newton in here.
I think this also looks very simple, although not as simple as the previous solution, with the advantage that it won’t generate false height changes as much as previous solution.
The time complexity of this solution is O(n^2)
, but if we had used some sort of sorted search tree with O(\log n)
operations instead of using lists, this could easily be improved to O(n \log n)
. After all what we do here is one simple iteration with foldl
, and then adding each point exactly once, and removing each end point at most once.
Note that height xs y
can also be implemented in O(\log n)
if xs is some kind of sorted search tree.
Solution 3. Divide and Conquer
Since this is functional programming and solutions usually tend to be solved using recursion, we could use a solution similar to merge sort. That is, solve the shape of skyline for each half of the buildings, and then merge the shapes.
The solution will look like:
1 2 3 4 5 6 7 8 9 10 11 |
skyline [] = [] skyline [(x1, h, x2)] = [(x1, h), (x2, 0)] skyline bs = (skyline (take n bs), 0) `merge` (skyline (drop n bs), 0) where n = (length bs) `div` 2 merge ([], _) (ys, _) = ys merge (xs, _) ([], _) = xs merge ((x, xh):xs, xh_p) ((y, yh):ys, yh_p) | x < y = (x, max xh yh_p) : merge (xs, xh) ((y, yh):ys, yh_p) | x == y = (x, max xh yh) : merge (xs, xh) (ys, yh) | x > y = (y, max xh_p yh) : merge ((x, xh):xs, xh_p) (ys, yh) |
The merge
function doesn’t seem very simple, so I wouldn’t use this solution if I could use a simpler solution.
Similar to merge sort, the time complexity of this solution is O(n \log n)
.
Conclusion
There are other solutions to this problem too, but I think these three solutions are good representative of the solution space. I love problems which can be solved in several totally different ways, and I found it beautiful that some of the solutions are strikingly simple and easy to understand.