Pergunta de entrevista da empresa Tesla

Python DP question, not too bad