Pergunta de entrevista da empresa IBM

Solving a DP question in python