Pregunta de entrevista de Tesla

Python DP question, not too bad