結果
問題 |
No.1239 Multiplication -2
|
ユーザー |
|
提出日時 | 2020-09-26 02:41:38 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 300 ms / 2,000 ms |
コード長 | 392 bytes |
コンパイル時間 | 162 ms |
コンパイル使用メモリ | 82,688 KB |
実行使用メモリ | 104,704 KB |
最終ジャッジ日時 | 2024-06-28 11:14:59 |
合計ジャッジ時間 | 6,740 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 34 |
ソースコード
mod = 998244353 N = int(input()) A = list(map(int, input().split())) ans = 0 dp = [0]*5 dp[1] = 1 for i in range(N): new_dp = [0]*5 new_dp[1] = pow(2,i,mod) for j in range(-2,3): if abs(j*A[i])<=2: new_dp[j*A[i]] += dp[j] dp = new_dp if i==N-1: ans += dp[-2] else: ans += dp[-2]*pow(2,N-2-i,mod) ans %= mod ans *= pow(pow(2,N-1,mod),mod-2,mod) print(ans%mod)