結果
問題 |
No.1741 Arrays and XOR Procedure
|
ユーザー |
|
提出日時 | 2021-10-15 19:49:50 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 832 bytes |
コンパイル時間 | 293 ms |
コンパイル使用メモリ | 82,612 KB |
実行使用メモリ | 111,244 KB |
最終ジャッジ日時 | 2024-11-25 10:56:55 |
合計ジャッジ時間 | 5,022 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 WA * 1 |
other | AC * 24 WA * 17 |
ソースコード
def combmod2(n, r): return (n & r) != r def solve(n, a): dp = [[0] * 2 for _ in range(n + 1)] dp[0][0] = 1 mod = 998244353 for i in range(n): if a[i] != -1: if combmod2(n - 1, i): dp[i + 1][1] = dp[i][1] dp[i + 1][0] = dp[i][0] else: dp[i + 1][1] += dp[i][(1 + a[i]) % 2] dp[i + 1][a[i]] += dp[i][a[i] % 2] else: if combmod2(n - 1, i): dp[i + 1][1] += 2 * dp[i][1] dp[i + 1][0] += 2 * dp[i][0] else: dp[i + 1][1] += dp[i][0] + dp[i][1] dp[i + 1][0] += dp[i][0] + dp[i][1] dp[i + 1][1] %= mod dp[i + 1][0] %= mod return dp[n][1] n = int(input()) a = list(map(int,input().split())) print(solve(n, a))