Pregunta de entrevista de IBM

Solving a DP question in python