結果
問題 |
No.1706 Many Bus Stops (hard)
|
ユーザー |
|
提出日時 | 2022-03-24 23:16:12 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 52 ms / 2,000 ms |
コード長 | 885 bytes |
コンパイル時間 | 166 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 62,592 KB |
最終ジャッジ日時 | 2024-10-13 06:57:45 |
合計ジャッジ時間 | 3,665 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 41 |
ソースコード
from copy import deepcopy MOD = 10 ** 9 + 7 def matpow(A, B, w): l = len(A) while w: if w & 1: C = [0] * l for i in range(l): for j in range(l): C[i] += A[i][j] * B[j] C[i] %= MOD B = C.copy() C = [[0] * l for _ in range(l)] for i in range(l): for j in range(l): for k in range(l): C[i][j] += A[i][k] * A[k][j] C[i][j] %= MOD A = deepcopy(C) w >>= 1 return B 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 -= 1 B = matpow(A, B, n) ans = 1 - B[0] ans = pow(ans, c, MOD) print((1 - ans) % MOD)