結果

問題 No.891 隣接3項間の漸化式
ユーザー kurimupy
提出日時 2020-06-15 19:53:26
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 518 bytes
コンパイル時間 81 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 44,736 KB
最終ジャッジ日時 2024-07-03 11:32:03
合計ジャッジ時間 22,142 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 7 WA * 28 RE * 4
権限があれば一括ダウンロードができます

ソースコード

diff #

#Fibonacchi sequence
import numpy as np
a,b,N=map(int,input().split())
A = np.array([[a,b], [1, 0]],dtype="int64")

mod =10 **9 + 7
def mat_f(X):
    if X == 0:
        return np.array([a, b], [0, 1])
    elif X == 1:
        return A
    else:
        if X % 2 == 0:
            S = mat_f(X//2)
            S%=mod
            return np.dot(S, S)
        else:
            S = mat_f(X//2)
            S%=mod
            return np.dot(np.dot(S, S), A)


mat = np.dot(mat_f(N), np.array([1, 0]))

b=mat[1]%mod
print(b)

0