Wentao Wang's submission of Graph Challenge | CodeChef