pramod's submission of INFECTED TRAIN | CodeChef