結果
問題 |
No.1239 Multiplication -2
|
ユーザー |
![]() |
提出日時 | 2020-09-03 22:26:43 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 735 ms / 2,000 ms |
コード長 | 475 bytes |
コンパイル時間 | 231 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 21,580 KB |
最終ジャッジ日時 | 2024-11-24 04:23:19 |
合計ジャッジ時間 | 13,225 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
ソースコード
mod = 998244353 inv = (mod + 1) // 2 n = int(input()) a = list(map(int, input().split())) dp = [0] * 5 dp[3] = 1 p = 1 pp = pow(2, n - 1, mod) ans = 0 for i in a: nex = [0] * 5 nex[3] = p p = p * 2 % mod ans = (ans + dp[0] * pp) % mod pp = pp * inv % mod for j in range(5): k = (j - 2) * i + 2 if 0 <= k <= 4: nex[k] = (nex[k] + dp[j]) % mod dp = nex ans = (ans + dp[0]) % mod print(ans * pow(2, mod - n, mod) % mod)