結果
問題 | No.995 タピオカオイシクナーレ |
ユーザー |
|
提出日時 | 2022-04-16 11:03:53 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 121 ms / 2,000 ms |
コード長 | 644 bytes |
コンパイル時間 | 524 ms |
コンパイル使用メモリ | 81,920 KB |
実行使用メモリ | 77,056 KB |
最終ジャッジ日時 | 2024-12-25 15:48:18 |
合計ジャッジ時間 | 3,749 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 23 |
ソースコード
n,m,k,p,q = map(int,input().split()) B = [int(input()) for i in range(n)] mod = 10**9+7 def calc(x,y): ans = [[0]*2 for i in range(2)] for i in range(2): for j in range(2): for t in range(2): ans[i][j] += x[i][t]*y[t][j] ans[i][j] %= mod return ans base = [[1,0],[0,1]] invq = pow(q,mod-2,mod) A = [[(q-p)*invq%mod,p*invq%mod],[p*invq%mod,(q-p)*invq%mod]] while k: if k & 1: base = calc(A,base) k >>= 1 A = calc(A,A) ans = 0 for i,b in enumerate(B): if i < m: ans += b*base[0][0] else: ans += b*base[0][1] ans %= mod print(ans)