codejam round 1 code