2/2

PKU 3744

ある場所からnステップ先の場所に行ける確率をf(n)とすると明らかに次の式が成り立つ。

f(n) = pf(n-1) + (1-p)f(n-2)

これの特性方程式

x^2 - px - 1 + p = 0

これを解くと

x = 1, p-1

よって

f(n) = c1 + c2(p-1)n

と書ける。f(0) = 1, f(1) = p は明らかなので

c1 + c2 = 1

c1 + c2(p-1) = p

これを連立して解けば

(c1, c2) = (1/(2-p), (p-1)/(p-2))

したがって

f(n) = ((p-1)n+1-1)/(p-2)

これが分かれば f(n) は O(log n) で計算できる(累乗の計算)ので後はやるだけ。