結果

問題 No.1595 The Final Digit
ユーザー osm ibtosm 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 40 ms
51,712 KB
testcase_01 AC 43 ms
51,840 KB
testcase_02 AC 44 ms
51,712 KB
testcase_03 AC 39 ms
51,584 KB
testcase_04 AC 45 ms
58,368 KB
testcase_05 AC 49 ms
58,112 KB
testcase_06 AC 46 ms
60,032 KB
testcase_07 AC 46 ms
59,904 KB
testcase_08 AC 40 ms
51,840 KB
testcase_09 AC 40 ms
52,352 KB
testcase_10 AC 41 ms
52,224 KB
testcase_11 AC 40 ms
52,096 KB
testcase_12 AC 41 ms
52,352 KB
testcase_13 AC 41 ms
52,096 KB
testcase_14 AC 48 ms
59,776 KB
testcase_15 AC 46 ms
59,776 KB
testcase_16 AC 47 ms
59,904 KB
testcase_17 AC 38 ms
51,968 KB
testcase_18 AC 39 ms
52,096 KB
testcase_19 AC 45 ms
60,032 KB
権限があれば一括ダウンロードができます

ソースコード

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