Solve problem 101 in Python
parent
5f32a5f18a
commit
2065f48a1a
|
@ -0,0 +1,37 @@
|
||||||
|
import numpy as np
|
||||||
|
|
||||||
|
|
||||||
|
def rev(xs):
|
||||||
|
return list(reversed(xs))
|
||||||
|
|
||||||
|
|
||||||
|
def bop(degree, f):
|
||||||
|
|
||||||
|
if degree == 1:
|
||||||
|
return 1
|
||||||
|
A = [rev([n ** x for x in range(degree)])
|
||||||
|
for n in range(1, degree + 1)]
|
||||||
|
y = np.array([f(n) for n in range(1, degree + 1)])
|
||||||
|
x = np.linalg.solve(A, y)
|
||||||
|
ns = rev([(degree + 1) ** x for x in range(degree)])
|
||||||
|
return np.dot(x, ns)
|
||||||
|
|
||||||
|
|
||||||
|
def euler_101():
|
||||||
|
# max_degree = 4
|
||||||
|
# def u(x):
|
||||||
|
# return x * x * x
|
||||||
|
|
||||||
|
max_degree = 10
|
||||||
|
def u(n):
|
||||||
|
return 1 - n + n**2 - n**3 + n**4 - n**5 + \
|
||||||
|
n**6 - n**7 + n**8 - n**9 + n**10
|
||||||
|
|
||||||
|
return int(sum([bop(d, u) for d in range(1, max_degree + 1)]))
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
solution = euler_101()
|
||||||
|
print("e101.py: " + str(solution))
|
||||||
|
assert(solution == 37076114526)
|
||||||
|
|
Loading…
Reference in New Issue