I agree, it took me ages to figure out an algorithm that was efficient enough to pass the tests.