Samarth Gupta's submission of Partition the Graph | CodeChef