Pregunta de entrevista de Tesla

A coding question, not LC type.