結果
問題 | No.1448 和差算 |
ユーザー |
![]() |
提出日時 | 2021-03-31 14:33:21 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 43 ms / 2,000 ms |
コード長 | 666 bytes |
コンパイル時間 | 465 ms |
コンパイル使用メモリ | 81,968 KB |
実行使用メモリ | 55,164 KB |
最終ジャッジ日時 | 2024-12-14 19:13:30 |
合計ジャッジ時間 | 2,882 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 36 |
ソースコード
M = [ [1, -1], [1, 1] ] MOD = 10 ** 9 + 7 def mul(a, b): return [[sum(a[i][k] * b[k][j] for k in range(2)) % MOD for j in range(2)] for i in range(2)] def matpow(x): ret = [[1, 0], [0, 1]] t = [m[:] for m in M] while x: if x & 1: ret = mul(ret, t) x >>= 1 t = mul(t, t) return ret # for i in range(10): # print(i) # print(*matpow(i), sep='\n', end='\n') A, B, C, D = map(int, input().split()) N = int(input()) if 3 <= N % 8 <= 6: a = A else: a = B if 2 <= N % 8 <= 5: b = C else: b = D X = matpow(N) ans = ((X[0][0] + X[1][0]) * a + (X[0][1] + X[1][1]) * b) % MOD print(ans)