結果

問題 No.2040 010-1 Deletion
ユーザー SSRSSSRS
提出日時 2022-08-12 21:59:02
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 107 ms / 3,000 ms
コード長 1,150 bytes
コンパイル時間 1,702 ms
コンパイル使用メモリ 176,116 KB
実行使用メモリ 109,184 KB
最終ジャッジ日時 2024-09-23 02:26:37
合計ジャッジ時間 4,926 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
const long long MOD = 998244353;
int main(){
  int N;
  cin >> N;
  string S;
  cin >> S;
  int r = -1;
  for (int i = 0; i < N; i++){
    if (S[i] == '0'){
      r = i;
    }
  }
  vector<vector<vector<long long>>> dp(N + 1, vector<vector<long long>>(2, vector<long long>(N * 3 + 1, 0)));
  dp[0][1][N] = 1;
  long long ans = 0;
  for (int i = 0; i < N; i++){
    for (int j = 0; j < 2; j++){
      for (int k = 0; k <= N * 3; k++){
        if (dp[i][j][k] > 0){
          if (S[i] != '1'){
            dp[i + 1][0][k - 1] += dp[i][j][k];
            dp[i + 1][0][k - 1] %= MOD;
            if (k - 1 >= N && (k - 1 - N) % 2 == 0 && i >= r){
              ans += dp[i][j][k];
              ans %= MOD;
            }
          }
          if (S[i] != '0'){
            if (j == 0){
              dp[i + 1][1][k + 2] += dp[i][j][k];
              dp[i + 1][1][k + 2] %= MOD;
            } else {
              dp[i + 1][1][k] += dp[i][j][k];
              dp[i + 1][1][k] %= MOD;
            }
          }
        }
      }
    }
  }
  if (r == -1){
    ans++;
    ans %= MOD;
  }
  cout << ans << endl;
}
0