manasa and stones hacker rank solution in python