Pregunta de entrevista de Amazon

I dont know how to write the second code.