結果
問題 | No.1706 Many Bus Stops (hard) |
ユーザー |
![]() |
提出日時 | 2021-10-08 22:48:27 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 52 ms / 2,000 ms |
コード長 | 1,090 bytes |
コンパイル時間 | 179 ms |
コンパイル使用メモリ | 82,580 KB |
実行使用メモリ | 61,568 KB |
最終ジャッジ日時 | 2024-07-23 05:55:05 |
合計ジャッジ時間 | 3,453 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 41 |
ソースコード
mod = 1000000007 eps = 10**-9 def main(): import sys input = sys.stdin.readline def matmul(A, B): C = [[0] * len(B[0]) for _ in range(len(A))] for i in range(len(A)): for k in range(len(B)): for j in range(len(B[0])): C[i][j] = (C[i][j] + A[i][k] * B[k][j]) % mod return C def matpow(A, p): n = len(A) B = [[0] * n for _ in range(n)] for i in range(n): B[i][i] = 1 while p > 0: if p & 1: B = matmul(B, A) A = matmul(A, A) p >>= 1 return B C, N, M = map(int, input().split()) invC = pow(C, mod - 2, mod) mat = [[0] * 4 for _ in range(4)] mat[0][0] = invC mat[0][3] = (1 - invC)%mod mat[1][1] = invC mat[1][2] = invC mat[1][3] = (1 - 2*invC)%mod mat[2][0] = 1 mat[3][1] = 1 vec = [[0] for _ in range(4)] vec[0][0] = 1 p = (1 - matmul(matpow(mat, N), vec)[0][0])%mod print((1 - pow(p, M, mod))%mod) if __name__ == '__main__': main()