Domanda di colloquio di Rakuten

A algorithm problem, medium difficulty