Question d’entretien chez IBM

Solving a DP question in python