Protium Interview Question

An easy level python coding question.