Pregunta de entrevista de Bending Spoons

An algorithm design programming question, including a final step with optimisation + explanation of my solution.