Yash Chandnani's submission of Partition the Graph | CodeChef