結果

問題 No.1654 Binary Compression
ユーザー SSRS
提出日時 2021-08-20 22:05:23
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 915 bytes
コンパイル時間 1,653 ms
コンパイル使用メモリ 171,404 KB
実行使用メモリ 66,888 KB
最終ジャッジ日時 2024-10-14 03:46:37
合計ジャッジ時間 11,896 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 8 WA * 42
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
const long long MOD = 924844033;
int main(){
  string S;
  cin >> S;
  int N = S.size();
  vector<int> p;
  for (int i = 0; i < N; i++){
    if (S[i] == '1'){
      p.push_back(i);
    }
  }
  vector<long long> dp0(N + 1, 0);
  vector<long long> dp1(N + 1, 0);
  dp0[0] = 1;
  for (int i = 0; i < N; i++){
    long long s = (dp0[i] + dp1[i]) % MOD;
    if (S[i] == '0'){
      dp0[i + 1] += s;
      dp0[i + 1] %= MOD;
      auto itr = lower_bound(p.begin(), p.end(), i);
      if (itr != p.end()){
        dp1[*itr + 1] += s;
        dp1[*itr + 1] %= MOD;
      }
    }
    if (S[i] == '1'){
      dp1[i + 1] += s;
      dp1[i + 1] %= MOD;
    }
  }
  long long ans = 0;
  for (int i = 1; i <= N; i++){
    ans += dp1[i];
  }
  for (int i = N - 1; i >= 0; i--){
    if (S[i] == '1'){
      break;
    }
    ans += dp0[i + 1];
  }
  ans %= MOD;
  cout << ans << endl;
}
0