Wentao Wang's submission of Short | CodeChef