Pregunta de entrevista de TD

a simple algorithm coding problem