Zhang Yide's submission of Optimal Partition | CodeChef