Zhenyang Dai's submission of Andrew and the Balls | CodeChef