結果
問題 | No.1706 Many Bus Stops (hard) |
ユーザー |
![]() |
提出日時 | 2021-10-08 23:10:42 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 706 bytes |
コンパイル時間 | 201 ms |
コンパイル使用メモリ | 81,992 KB |
実行使用メモリ | 63,224 KB |
最終ジャッジ日時 | 2024-07-23 06:46:31 |
合計ジャッジ時間 | 3,536 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 1 WA * 40 |
ソースコード
def mmult(A, B):global modn, 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]) % modreturn retdef mpow(A, n):if n == 1: return Aif 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 + 7iv = pow(C, mod - 2, mod)ivv = iv * (C - 2) % modX = [[iv, 0, 0, iv], [0, iv, iv, ivv], [1, 0, 0, 0], [0, 1, 0, 0]]print((1 - pow(1 - mpow(X, N)[0][0], M, mod)) % mod)