def F(n,p):
"""
n 本の傘を持っているとする。
n + 1 本目の置き傘ができるまでの移動回数の期待値を求める。
"""
import numpy as np
q = 1 - p
A = np.eye(n+1)
A[np.arange(n+1), np.arange(n+1)[::-1]] -= q
A[np.arange(1,n+1), np.arange(1,n+1)[::-1]] -= p
x = np.linalg.solve(A, np.ones(n+1))
return x[-1]
N = 10
p = 0.3
E = [F(n,p) for n in range(N)]
print(E)
answer = sum(E)
print(answer)