Pregunta de entrevista de Lyft

a simple coding exercise in Python