結果

問題 No.1595 The Final Digit
ユーザー dachengz
提出日時 2022-12-17 13:53:18
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 45 ms / 2,000 ms
コード長 606 bytes
コンパイル時間 711 ms
コンパイル使用メモリ 82,140 KB
実行使用メモリ 59,392 KB
最終ジャッジ日時 2024-11-16 22:08:25
合計ジャッジ時間 2,179 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

def matrixmul(A, B):
    C = [[0] * 3 for _ in range(3)]
    for i, row in enumerate(A):
        Ci = C[i]
        for j, vA in enumerate(row):
            for k, vB in enumerate(B[j]):
                Ci[k] = (Ci[k] + vA * vB) % 10
    return C

def matrixpow(A, e):
    R = [[1, 0, 0], [0, 1, 0], [0, 0, 1]]
    while e > 0:
        if e & 1:
            R = matrixmul(R, A)
        e >>= 1
        A = matrixmul(A, A)
    return R

p, q, r, K = map(int, input().split())
A = [[1, 1, 1], [1, 0, 0], [0, 1, 0]]
init = [[r % 10], [q % 10], [p % 10]]
B = matrixmul(matrixpow(A, K - 3), init)
print(B[0][0])
0