Zhang Yide's submission of Lights | CodeChef