結果
問題 |
No.1239 Multiplication -2
|
ユーザー |
![]() |
提出日時 | 2020-09-03 22:06:48 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 814 ms / 2,000 ms |
コード長 | 569 bytes |
コンパイル時間 | 274 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 21,720 KB |
最終ジャッジ日時 | 2024-11-24 03:48:57 |
合計ジャッジ時間 | 15,030 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, max(0, n - 2), mod) ans = 0 for i in range(n): nex = [0] * 5 for j in range(5): k = (j - 2) * a[i] + 2 if not (0 <= k <= 4): k = 2 nex[k] = (nex[k] + dp[j]) % mod nex[3] = (nex[3] + p) % mod p = p * 2 % mod dp = nex if i < n - 1: ans = (ans + dp[0] * pp) % mod else: ans = (ans + dp[0]) % mod pp = pp * inv % mod print(ans * pow(2, mod - n, mod) % mod)