結果
問題 | No.1595 The Final Digit |
ユーザー | wolgnik |
提出日時 | 2021-07-09 21:52:07 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 50 ms / 2,000 ms |
コード長 | 712 bytes |
コンパイル時間 | 234 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 60,288 KB |
最終ジャッジ日時 | 2024-07-01 16:06:28 |
合計ジャッジ時間 | 1,809 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 17 |
ソースコード
import sys input = sys.stdin.readline p, q, r, K = map(int, input().split()) def ArrayMultiple(a, b, mod): n = len(a) m = len(b[0]) c = [[0] * n for _ in range(n)] for i in range(n): for j in range(m): for k in range(n): c[i][j] += a[i][k] * b[k][j] c[i][j] %= mod return c def ArrayPower(a, k, mod): bn = list(bin(k)[2: ]) ln = len(bn) n = len(a) bn.reverse() res = [[int(i == x) for i in range(n)] for x in range(n)] for i in range(ln): if int(bn[i]): res = ArrayMultiple(res, a, mod) a = ArrayMultiple(a, a, mod) return res a = [[1, 1, 1], [1, 0, 0], [0, 1, 0]] ak = ArrayPower(a, K - 3, 10)[0] print((ak[0] * r + ak[1] * q + ak[2] * p) % 10)