結果
問題 |
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+7 cinv = 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] %= mod return ans base = [[0]*4 for i in range(4)] for i in range(4): base[i][i] = 1 while n: if n%2: base = calc(A,base) n //= 2 A = calc(A,A) p = (base[2][0]*cinv+base[2][2])%mod ans = (1-pow(1-p,m,mod))%mod print(ans)