In the description you said:

Your solution should be efficient enough to not throw a timeout exception.

but your tests are very simple and not really test the efficiency of the algorithm!