結果

問題 No.1595 The Final Digit
ユーザー osm ibt
提出日時 2021-07-10 12:49:58
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 49 ms / 2,000 ms
コード長 748 bytes
コンパイル時間 148 ms
コンパイル使用メモリ 82,244 KB
実行使用メモリ 60,032 KB
最終ジャッジ日時 2024-07-02 01:53:11
合計ジャッジ時間 1,732 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 


def prod(ma, mb, mod = 10**9+7):
    n = len(ma)
    ret = [[0]*n for _ in range(n)]
    for i in range(n):
        for j in range(n):
            for k in range(n):
                ret[i][j] += ma[i][k]*mb[k][j]
                ret[i][j] %= mod
    return ret

def powmat(ma, k, mod = 10**9+7):
    n = len(ma)
    ret = [[0]*n for _ in range(n)]
    for i in range(n):
        ret[i][i] = 1
    for _ in range(k):
        if k & 1:
            ret = prod(ret, ma, mod)
        ma = prod(ma, ma, mod)
        k >>= 1
        if k == 0: break
    return ret

p, q, r, k = map(int, input().split())

A = [[1, 1, 1], [1, 0, 0], [0, 1, 0]]
Ak = powmat(A, k-3, 10)[0]

ret = 0
ret += Ak[0] * r
ret += Ak[1] * q
ret += Ak[2] * p
ret %= 10

print(ret)
0