Sajal Gupta's submission of ATM | CodeChef