結果

問題 No.1741 Arrays and XOR Procedure
コンテスト
ユーザー eve__fuyuki
提出日時 2024-04-10 00:59:16
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 46 ms / 2,000 ms
コード長 1,295 bytes
コンパイル時間 2,229 ms
コンパイル使用メモリ 200,244 KB
最終ジャッジ日時 2025-02-20 23:36:40
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

#include <atcoder/modint>
using namespace std;
using mint = atcoder::modint998244353;
void fast_io() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
}

int main() {
    fast_io();
    int n;
    cin >> n;
    vector<int> b(n);
    for (int i = 0; i < n; i++) {
        cin >> b[i];
    }
    vector<int> cnt(n);
    for (int i = 1; i < n; i++) {
        int p = 2;
        while (p <= i) {
            cnt[i] += i / p;
            p *= 2;
        }
    }
    vector<vector<mint>> dp(n + 1, vector<mint>(2, 0));
    dp[0][0] = 1;
    for (int i = 0; i < n; i++) {
        if (cnt[n - 1] > cnt[i] + cnt[n - 1 - i]) {
            if (b[i] == -1) {
                dp[i + 1][0] = 2 * dp[i][0];
                dp[i + 1][1] = 2 * dp[i][1];
            } else {
                dp[i + 1][0] = dp[i][0];
                dp[i + 1][1] = dp[i][1];
            }
        } else {
            if (b[i] == -1) {
                dp[i + 1][0] = dp[i + 1][1] = dp[i][0] + dp[i][1];
            } else if (b[i] == 0) {
                dp[i + 1][0] = dp[i][0];
                dp[i + 1][1] = dp[i][1];
            } else {
                dp[i + 1][0] = dp[i][1];
                dp[i + 1][1] = dp[i][0];
            }
        }
    }
    cout << dp[n][1].val() << endl;
}
0