manasa and stones hackerrank solution in python