Source code about codejam problem