三角数,五角数和六角数分别通过以下公式定义:
三角数 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)