Shreyas Gupta's submission of ATM | CodeChef