結果

問題 No.2279 OR Insertion
ユーザー miscalc
提出日時 2023-04-19 14:04:10
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,017 bytes
コンパイル時間 1,924 ms
コンパイル使用メモリ 202,904 KB
最終ジャッジ日時 2025-02-12 10:19:21
ジャッジサーバーID
(参考情報)
judge3 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 18 TLE * 30
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#include <atcoder/modint>
using namespace atcoder;
using mint = modint998244353;

#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")

int main()
{
  int N;
  string S;
  cin >> N >> S;

  mint ans = 0;
  vector<ll> dp(N + 1, 0);
  for (ll k = 0; k < N; k++)
  {
    dp.at(0) = 1;
    for (int i = 1; i <= N; i++)
    {
      dp.at(i) = 0;
    }

    for (int i = 0; i < N; i++)
    {
      if (i - k < 0 || S.at(i - k) == '0')
      {
        for (int j = 0; j <= i; j++)
        {
          dp[i + 1] += dp[j];
          if (dp[i + 1] >= mint::mod())
            dp[i + 1] -= mint::mod();
        }
      }
      else
      {
        for (int j = i - k + 1; j <= i; j++)
        {
          dp[i + 1] += dp[j];
          if (dp[i + 1] >= mint::mod())
            dp[i + 1] -= mint::mod();
        }
      }
    }
    ans += (mint(2).pow(N - 1) - dp.at(N)) * mint(2).pow(k);
  }
  cout << ans.val() << endl;
}
0