結果
問題 | No.1706 Many Bus Stops (hard) |
ユーザー |
![]() |
提出日時 | 2021-10-08 23:21:21 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 707 bytes |
コンパイル時間 | 158 ms |
コンパイル使用メモリ | 82,116 KB |
実行使用メモリ | 62,944 KB |
最終ジャッジ日時 | 2024-07-23 07:17:16 |
合計ジャッジ時間 | 3,475 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | WA * 41 |
ソースコード
def mmult(A, B): global mod n, m, l = len(A), len(B), len(B[0]) ret = [[0] * l for _ in range(n)] for i in range(n): for j in range(m): for k in range(l): ret[i][k] = (ret[i][k] + A[i][j] * B[j][k]) % mod return ret def mpow(A, n): if n == 1: return A if n == 0: return [[1 if i == j else 0 for j in range(len(A))] for i in range(len(A))] return mmult(mpow(A, n - 1), A) if n % 2 else mpow(mmult(A, A), n // 2) C, N, M = map(int, input().split()) mod = 10 ** 9 + 7 iv = pow(C, mod - 2, mod) ivv = iv * (C - 2) % mod X = [[iv, 0, 0, iv], [0, iv, ivv, ivv], [1, 0, 0, 0], [0, 1, 0, 0]] print((1 - pow(1 - mpow(X, N)[0][0], M, mod)) % mod)