結果
問題 | No.444 旨味の相乗効果 |
ユーザー |
![]() |
提出日時 | 2021-02-05 08:01:10 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 75 ms / 2,500 ms |
コード長 | 635 bytes |
コンパイル時間 | 210 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 76,056 KB |
最終ジャッジ日時 | 2024-07-01 20:45:18 |
合計ジャッジ時間 | 2,420 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 5 |
other | AC * 23 |
ソースコード
def polymul(f,g):lf = len(f)lg = len(g)res = [0]*(lf+lg-1)for i in range(lf):for j in range(lg):res[i+j] += f[i]*g[j]res[i+j] %= MODreturn resdef fps_nth_term(f,g,N):assert g[0] != 0while N:h = g[:]for i in range(1,len(g),2):h[i] = -h[i]f = polymul(f,h)[N%2:N+1:2]g = polymul(g,h)[:N+1:2]N //= 2return f[0]*pow(g[0],MOD-2,MOD)%MODn,c,*a = map(int,open(0).read().split())MOD = 10**9+7r = [1]x = 0for ai in a:r = polymul(r,[1,-ai])x -= pow(ai,c,MOD)print((x+fps_nth_term([1],r,c))%MOD)