結果

問題 No.2279 OR Insertion
ユーザー magstamagsta
提出日時 2023-04-19 14:18:38
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 883 bytes
コンパイル時間 1,804 ms
コンパイル使用メモリ 172,892 KB
実行使用メモリ 10,152 KB
最終ジャッジ日時 2024-10-14 14:26:40
合計ジャッジ時間 54,205 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 32 TLE * 16
権限があれば一括ダウンロードができます

ソースコード

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;
  ll dp[4010];
  for (ll k = 0; k < N; k++)
  {
    dp[0] = 1;
    for (int i = 1; i <= N; i++)
    {
      dp[i] = 0;
    }

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