Itertools Hackerrank Solution