結果
問題 | No.1448 和差算 |
ユーザー |
|
提出日時 | 2021-03-31 15:00:34 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 48 ms / 2,000 ms |
コード長 | 733 bytes |
コンパイル時間 | 892 ms |
コンパイル使用メモリ | 82,332 KB |
実行使用メモリ | 53,248 KB |
最終ジャッジ日時 | 2024-12-14 22:36:12 |
合計ジャッジ時間 | 2,958 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 36 |
ソースコード
A, B, C, D = map(int, input().split())N = int(input())MOD = 10 ** 9 + 7if (N + 1) % 8 <= 4:a0 = Belse:a0 = Aif (N - 1) % 8 <= 4:b0 = Celse:b0 = Ddef matmul(X, Y):ret = [[0] * 2 for i in range(2)]for i in range(2):for j in range(2):for k in range(2):ret[i][j] = (ret[i][j] + X[i][k] * Y[k][j]) % MODreturn retdef matpow(X, n):ret = [[0] * 2 for i in range(2)]ret[0][0] = 1ret[1][1] = 1while n > 0:if n % 2:ret = matmul(ret, X)X = matmul(X, X)n >>= 1return retret = matpow([[1, -1],[1, 1]], N)ans = ret[0][0] * a0 + ret[0][1] * b0 + ret[1][0] * a0 + ret[1][1] * b0print(ans % MOD)