Pentagon Numbers#
from itertools import *
from math import *
from operator import *
def pentagonal(n):
return n * (3 * n - 1) // 2
pentagonals = set(pentagonal(n) for n in range(1, 3000))
c = combinations(pentagonals, 2)
for p in c:
if add(*p) in pentagonals and abs(sub(*p)) in pentagonals:
print(abs(sub(*p)))
OUTPUT
5482660