結果
問題 |
No.526 フィボナッチ数列の第N項をMで割った余りを求める
|
ユーザー |
|
提出日時 | 2020-07-14 10:03:21 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 29 ms / 2,000 ms |
コード長 | 879 bytes |
コンパイル時間 | 204 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 10,880 KB |
最終ジャッジ日時 | 2024-11-16 20:36:03 |
合計ジャッジ時間 | 1,450 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 12 |
ソースコード
def power(x,n,mod): digit = len(str(bin(n)))-2 a = len(x) dp = [x]*digit for i in range(1,digit): dp[i] = dot(dp[i-1],dp[i-1],mod) ans = [[0]*a for _ in range(a)] for i in range(a): ans[i][i] = 1 for i in range(digit): if 1<<i & n: ans = dot(dp[i],ans,mod) return ans def dot(A,B,mod): p = len(A) q = len(A[0]) r = len(B) s = len(B[0]) if q!=r: print('掛け算できません') return ans = [[0]*s for _ in range(p)] for i in range(p): for j in range(s): for k in range(q): ans[i][j] += A[i][k]*B[k][j] ans[i][j] %= mod return ans def solve(): ans = 0 N, M = map(int, input().split()) init = [[1],[0]] mat = [[1,1],[1,0]] ans = dot(power(mat,N-2,M),init,M) return ans[0][0] print(solve())