結果

問題 No.891 隣接3項間の漸化式
ユーザー realDivineJK
提出日時 2020-05-19 00:11:44
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 30 ms / 2,000 ms
コード長 778 bytes
コンパイル時間 318 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 10,880 KB
最終ジャッジ日時 2024-10-01 22:18:02
合計ジャッジ時間 2,583 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 39
権限があれば一括ダウンロードができます

ソースコード

diff # 

a, b, n = map(int, input().split())
mod = int(1e9)+7
vec = [0, 1]
mat = [[1, 0], [0, 1]]
bas = [[0, 1], [b, a]]
while n != 0:
    t = [[0, 0], [0, 0]]
    if n % 2 == 1:
        for i in range(2):
            for j in range(2):
                for k in range(2):
                    t[i][j] += mat[i][k] * bas[k][j]
                    t[i][j] %= mod
        for i in range(2):
            for j in range(2):
                mat[i][j] = t[i][j]
    for i in range(2):
        for j in range(2):
            t[i][j] = 0
            for k in range(2):
                t[i][j] += bas[i][k] * bas[k][j]
                t[i][j] %= mod
    for i in range(2):
        for j in range(2):
            bas[i][j] = t[i][j]
    n //= 2
print((mat[0][0] * vec[0] + mat[0][1] * vec[1]) % mod)
0