Jiaye Zhu's submission of Count the squares | CodeChef