結果

問題 No.1706 Many Bus Stops (hard)
ユーザー 👑 rin204
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

from copy import deepcopy
MOD = 10 ** 9 + 7
def 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 -= 1
while n:
if n & 1:
C = [0] * 4
for i in range(4):
for j in range(4):
C[i] += A[i][j] * B[j]
C[i] %= MOD
B = C.copy()
n >>= 1
C = [[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] %= MOD
A = deepcopy(C)
ans = 1 - B[0]
ans = pow(ans, c, MOD)
print((1 - ans) % MOD)
for _ in range(1):
solve()
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0