結果

問題 No.2279 OR Insertion
ユーザー Yilin Fei
提出日時 2024-01-08 18:23:04
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,023 bytes
コンパイル時間 769 ms
コンパイル使用メモリ 73,120 KB
実行使用メモリ 128,896 KB
最終ジャッジ日時 2024-09-27 19:38:47
合計ジャッジ時間 7,361 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 2
other AC * 3 WA * 45
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>

int main() {
    int N;
    std::cin >> N;

    std::vector<std::vector<long long>> A(N + 1, std::vector<long long>(N, 0));
    std::vector<long long> B(N, 0);
    std::string S;
    std::cin >> S;

    const int P = 998244353;
    std::vector<long long> p(N);
    
    p[0] = 1; // base case for powers of 2
    for (int i = 1; i < N; ++i) {
        p[i] = (2 * p[i - 1]) % P; // precalculate powers of 2 modulo P
    }

    for (int i = 1; i <= N; ++i) {
        bool s = S[N - i] > '0';
        for (int j = 0; j < i; ++j) {
            long long val = (s && j > 0) ? (p[std::max(0, i - j - 2)] - A[i - j - 1][j]) % P : 0;
            if (val < 0) val += P; // ensure non-negative value
            B[j] = (B[j] + val + (s && j == 0 ? p[std::max(0, i - 2)] : 0)) % P;
            A[i][j] = B[j];
        }
    }

    long long result = 0;
    for (int j = 0; j < N; ++j) {
        result = (result + p[j] * A[N][j]) % P;
    }

    std::cout << result << std::endl;
    return 0;
}
0