[交流] Euler 工程 第四十四题 已有3人参与

import math
p=[1]
n=1

def isPentagonal(a):
    b=(1+math.sqrt(1+24*a))/6
    if b-int(b)==0:
        return True
    else:
        return False

flag=True
while flag:
    n=n+1
    p.append(int(n*(3*n-1)/2))
    for i in range(n-2,-1,-1):
        n1=p[n-1]+p[i]
        n2=p[n-1]-p[i]
        if isPentagonal(n1) and isPentagonal(n2):
            print(n2)
            flag=False
            break