Rohit Goyal's submission of Partition the Graph | CodeChef