結果
問題 |
No.1239 Multiplication -2
|
ユーザー |
![]() |
提出日時 | 2020-09-03 22:07:23 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 127 ms / 2,000 ms |
コード長 | 593 bytes |
コンパイル時間 | 197 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 104,192 KB |
最終ジャッジ日時 | 2024-11-24 03:49:41 |
合計ジャッジ時間 | 4,504 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
ソースコード
#!/usr/bin/env python3 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)