Tiancheng Lou's submission of Partition the Graph | CodeChef