結果

問題 No.1595 The Final Digit
ユーザー neterukunneterukun
提出日時 2021-07-09 21:36:19
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 76 ms / 2,000 ms
コード長 1,122 bytes
コンパイル時間 322 ms
コンパイル使用メモリ 86,968 KB
実行使用メモリ 76,588 KB
最終ジャッジ日時 2023-09-14 08:00:50
合計ジャッジ時間 2,648 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge13
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 67 ms
71,448 KB
testcase_01 AC 69 ms
71,272 KB
testcase_02 AC 69 ms
71,432 KB
testcase_03 AC 69 ms
71,444 KB
testcase_04 AC 74 ms
75,568 KB
testcase_05 AC 74 ms
75,620 KB
testcase_06 AC 76 ms
76,388 KB
testcase_07 AC 76 ms
76,400 KB
testcase_08 AC 69 ms
71,476 KB
testcase_09 AC 68 ms
71,156 KB
testcase_10 AC 69 ms
71,480 KB
testcase_11 AC 70 ms
71,444 KB
testcase_12 AC 69 ms
71,444 KB
testcase_13 AC 69 ms
71,304 KB
testcase_14 AC 74 ms
76,048 KB
testcase_15 AC 75 ms
76,244 KB
testcase_16 AC 73 ms
76,588 KB
testcase_17 AC 69 ms
71,204 KB
testcase_18 AC 70 ms
71,560 KB
testcase_19 AC 74 ms
76,472 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

def matrix_pow(A, k, MOD):
    def matrix_mul(A, B, MOD):
        C = [[0] * size for _ in range(size)]
        for i in range(size):
            for k in range(size):
                for j in range(size):
                    C[i][j] = (C[i][j] + A[i][k] * B[k][j]) % MOD
        return C

    if len(A) != len(A[0]):
        raise ValueError('square matrix expected')

    size = len(A)
    B = [[0] * size for _ in range(size)]
    for i in range(size):
        B[i][i] = 1
    while k > 0:
        if k & 1:
            B = matrix_mul(A, B, MOD)
        A = matrix_mul(A, A, MOD)
        k = k // 2
    return B


def matvec_mul(A, vec, MOD):
    if len(A[0]) != len(vec):
        raise ValueError('dimension mismatch between matrix and vector')

    h, w = len(A), len(A[0])
    res_vec = [0] * h
    for i in range(h):
        for j in range(w):
            res_vec[i] += A[i][j] * vec[j]
            res_vec[i] %= MOD
    return res_vec


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


mat = [[1, 1, 1], [1, 0, 0], [0, 1, 0]]


mat = matrix_pow(mat, k - 3, MOD)
res = matvec_mul(mat, [r, q, p], MOD)
print(res[0])
0