Devansh Bajpai's submission of Train Partner | CodeChef