結果
問題 | No.1706 Many Bus Stops (hard) |
ユーザー |
|
提出日時 | 2021-10-09 17:27:51 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 47 ms / 2,000 ms |
コード長 | 611 bytes |
コンパイル時間 | 327 ms |
コンパイル使用メモリ | 82,356 KB |
実行使用メモリ | 62,128 KB |
最終ジャッジ日時 | 2024-09-13 03:29:08 |
合計ジャッジ時間 | 3,577 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 41 |
ソースコード
c,n,m = map(int,input().split())mod = 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)