結果

問題 No.526 フィボナッチ数列の第N項をMで割った余りを求める
ユーザー proribone
提出日時 2020-10-07 22:54:20
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 44 ms / 2,000 ms
コード長 552 bytes
コンパイル時間 225 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 52,608 KB
最終ジャッジ日時 2024-07-20 04:18:41
合計ジャッジ時間 1,665 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 12
権限があれば一括ダウンロードができます

ソースコード

diff #

def mul(A,B):
    C=[[0]*len(B) for i in range(len(A[0]))]
    for i in range(len(A)):
        for j in range(len(B[0])):
            for k in range(len(B)):
                C[i][j]=(C[i][j]+A[i][k]*B[k][j])%M
    return C

def matpow(vec,n):
    B=[[0]*len(vec) for i in range(len(vec))]
    for i in range(len(vec)):
        B[i][i]=1
    while(n>0):
        if n&1:
            B=mul(B,vec)
        vec=mul(vec,vec)
        n>>=1
    return B
    
N,M=map(int,input().split())
N-=1
A=[[1],[0]]
V=[[1,1],[1,0]]
B=matpow(V,N)
A=mul(B,A)
print(A[1][0])
0