結果
問題 | No.1706 Many Bus Stops (hard) |
ユーザー |
👑 |
提出日時 | 2022-01-23 01:40:50 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 45 ms / 2,000 ms |
コード長 | 907 bytes |
コンパイル時間 | 170 ms |
コンパイル使用メモリ | 82,040 KB |
実行使用メモリ | 63,632 KB |
最終ジャッジ日時 | 2024-11-28 14:00:36 |
合計ジャッジ時間 | 3,230 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 41 |
ソースコード
from copy import deepcopyMOD = 10 ** 9 + 7def solve():m, n, c = map(int, input().split())inv = pow(m, MOD - 2, MOD) % MOD# A_i, A_{i-1}, B_i, B_{i-1}B = [inv, 1, 0, 0]A = [[inv, 0, 0, inv],[1, 0, 0, 0],[0, (m - 1) * inv, inv, (m - 2) * inv],[0, 0, 1, 0]]n -= 1while n:if n & 1:C = [0] * 4for i in range(4):for j in range(4):C[i] += A[i][j] * B[j]C[i] %= MODB = C.copy()n >>= 1C = [[0] * 4 for _ in range(4)]for i in range(4):for j in range(4):for k in range(4):C[i][j] += A[i][k] * A[k][j]C[i][j] %= MODA = deepcopy(C)ans = 1 - B[0]ans = pow(ans, c, MOD)print((1 - ans) % MOD)for _ in range(1):solve()