Zhang Yide's submission of A Good Problem | CodeChef