pentagonal = [n*(3*n-1)/2 for n in xrange(1, 10000)] pentagonal_set = set(pentagonal) for j in pentagonal: for k in pentagonal: if j+k in pentagonal_set and j-k in pentagonal_set: print abs(j-k)