結果
問題 | No.1704 Many Bus Stops (easy) |
ユーザー |
|
提出日時 | 2021-10-09 17:29:18 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 304 ms / 2,000 ms |
コード長 | 737 bytes |
コンパイル時間 | 489 ms |
コンパイル使用メモリ | 82,372 KB |
実行使用メモリ | 78,696 KB |
最終ジャッジ日時 | 2024-09-13 03:31:23 |
合計ジャッジ時間 | 11,435 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 41 |
ソースコード
t = int(input())for _ in range(t):n = int(input())c,m = 3,1mod = 10**9+7cinv = pow(c,mod-2,mod)A = [[cinv,0,0,cinv],[0,cinv,(c-1)*cinv%mod,(c-2)*cinv%mod],[1,0,0,0],[0,1,0,0]]def calc(X,Y):ans = [[0]*4 for i in range(4)]for i in range(4):for j in range(4):for k in range(4):ans[i][j] += X[i][k]*Y[k][j]ans[i][j] %= modreturn ansbase = [[0]*4 for i in range(4)]for i in range(4):base[i][i] = 1while n:if n%2:base = calc(A,base)n //= 2A = calc(A,A)p = (base[2][0]*cinv+base[2][2])%modans = (1-pow(1-p,m,mod))%modprint(ans)