project-euler/078_coin_partitions.py

30 lines
524 B
Python
Raw Normal View History

2023-03-27 16:30:58 +02:00
#!/usr/bin/env python3
def main(d):
tbl = [1]
n = 1
while p(n, tbl) % d: n+=1
return n, p(n,tbl)
def p(n, tbl=[1]):
if n < 0:
return 0
if n >= len(tbl):
r = -sum((-1)**i *
(p(n-a, tbl) +
p(n-b, tbl))
for a,b,i in pentnum(n))
tbl.append(r)
return tbl[n]
def pentnum(bound):
n = 1
a,b = 1,2
while a <= bound:
yield a,b,n
a += 3*n + 1
b += 3*n + 2
n += 1
print(main(10**6)[0])