Pregunta de entrevista de Infosys

Write a program in python