Pergunta de entrevista da empresa Lyft

a simple coding exercise in Python