欧拉计划 45

三角数，五角数和六角数分别通过以下公式定义：

三角数 Tn=n(n+1)/2 1, 3, 6, 10, 15, …
五角数 Pn=n(3n−1)/2 1, 5, 12, 22, 35, …
六角数 Hn=n(2n−1) 1, 6, 15, 28, 45, …
可以证实 T285 = P165 = H143 = 40755.

找出这之后的下一个既是五角数又是六角数的三角数。

def T(n):
    """ 三角数 """
    return n * (n + 1) / 2


def P(n):
    """ 五角数 """
    return n * (3 * n - 1) / 2


def H(n):
    """ 六角数 """
    return n * (2 * n - 1)


i = 144
j = 166
k = 286
while 1:
    hn = H(i)
    while P(j) < hn:
        j += 1
    if P(j) == hn:
        while T(k) < hn:
            k += 1
        if T(k) == hn:
            break
    i += 1
print(hn)