Shubhi Rustagi's submission of ATM | CodeChef