結果
問題 |
No.1741 Arrays and XOR Procedure
|
ユーザー |
|
提出日時 | 2021-10-15 19:39:38 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 115 ms / 2,000 ms |
コード長 | 825 bytes |
コンパイル時間 | 140 ms |
コンパイル使用メモリ | 82,072 KB |
実行使用メモリ | 111,444 KB |
最終ジャッジ日時 | 2024-11-25 10:56:20 |
合計ジャッジ時間 | 5,133 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
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 + a[i]) % 2] += dp[i][1] dp[i + 1][a[i]] += dp[i][0] 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))