Zhang Yide's submission of ATM | CodeChef