結果
問題 | No.1704 Many Bus Stops (easy) |
ユーザー |
|
提出日時 | 2021-11-14 22:37:15 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 792 ms / 2,000 ms |
コード長 | 1,074 bytes |
コンパイル時間 | 370 ms |
コンパイル使用メモリ | 82,688 KB |
実行使用メモリ | 78,464 KB |
最終ジャッジ日時 | 2024-11-30 08:24:12 |
合計ジャッジ時間 | 23,120 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 41 |
ソースコード
mod = 10**9 + 7def matmul(A, B):Ah, Bh, Bw = len(A), len(B), len(B[0])C = [[0 for _ in range(Bw)] for _ in range(Ah)]for i in range(Ah):for j in range(Bw):for k in range(Bh):C[i][j] += A[i][k] * B[k][j]C[i][j] %= modreturn C# Mのk乗を効率的に計算するdef doubling(M, k):k -= 1Mc = M.copy()while k > 0:if k & 1 == 1:Mc = matmul(Mc, M)M = matmul(M, M) # Mの(2のi乗)の乗 を計算するk >>= 1return McA = [[1, 0, 0, 0, 1, 1],[0, 1, 0, 1, 0, 1],[0, 0, 1, 1, 1, 0],[3, 0, 0, 0, 0, 0],[0, 3, 0, 0, 0, 0],[0, 0, 3, 0, 0, 0]]F = [1, 0, 0, 3, 0, 0]fraction = pow(3, mod - 2, mod)t = int(input())for _ in range(t):n = int(input())if n == 1:print(fraction)else:M = doubling(A, n)ans = 0for i in range(6):ans += M[3][i] * F[i]ans %= modans *= pow(fraction, n + 1, mod)ans %= modprint(ans)