Pregunta de entrevista de Rakuten

A algorithm problem, medium difficulty