結果

問題 No.2541 Divide 01 String
ユーザー mkreemmkreem
提出日時 2024-01-20 11:35:55
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 22 ms / 2,000 ms
コード長 2,130 bytes
コンパイル時間 3,192 ms
コンパイル使用メモリ 250,600 KB
実行使用メモリ 14,424 KB
最終ジャッジ日時 2024-09-28 05:31:56
合計ジャッジ時間 3,720 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 5 ms
5,376 KB
testcase_04 AC 13 ms
8,832 KB
testcase_05 AC 20 ms
13,568 KB
testcase_06 AC 11 ms
7,936 KB
testcase_07 AC 21 ms
13,312 KB
testcase_08 AC 20 ms
14,336 KB
testcase_09 AC 22 ms
14,336 KB
testcase_10 AC 20 ms
14,424 KB
testcase_11 AC 21 ms
14,336 KB
testcase_12 AC 21 ms
14,336 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 2 ms
5,376 KB
testcase_15 AC 2 ms
5,376 KB
testcase_16 AC 2 ms
5,376 KB
testcase_17 AC 1 ms
5,376 KB
testcase_18 AC 14 ms
10,368 KB
testcase_19 AC 14 ms
10,368 KB
testcase_20 AC 14 ms
10,112 KB
testcase_21 AC 12 ms
8,576 KB
testcase_22 AC 12 ms
8,448 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#ifndef INCLUDED_main
#define INCLUDED_main

#include __FILE__

const ll mod = 998244353; using mint = atcoder::modint998244353;
//const ll mod = 1000000007; using mint = atcoder::modint1000000007;





int main(){

  int n; cin >> n;
  string s; cin >> s;

  vector dp(n+1, vector<mint>(2));
  dp[0][0] = 1;
  rep(i, 0, n){
    if(s[i] == '1'){
      dp[i+1][1] += dp[i][0] + dp[i][1]; // 分割する
      dp[i+1][1] += dp[i][1]; // しない
    }
    else{
      dp[i+1][0] += dp[i][0]; // しない
      dp[i+1][1] += dp[i][1];
      dp[i+1][0] += dp[i][1]; // する
    }
  }

  cout << dp[n][1].val() << endl;

}

#else 

#include <bits/stdc++.h>
#include <atcoder/modint>

using ll = long long;
using ld = long double;
using Graph = std::vector<std::vector<int>>;
#define endl '\n'
#define INF 1000000039
#define LLINF 393939393939393939
#define fore(i, a) for(auto &i : a)
#define rep(i, a, b) for(int i = a; i < b; i++)
#define erep(i, a, b) for(int i = a; i <= b; i++)
#define rrep(i, a, b) for(int i = a; i >= b; i--)
#define all(x) x.begin(), x.end()
#define rall(x) x.rbegin(), x.rend()
#define part(x, a, b) x.begin()+a, x.begin()+b+1
#define rpart(x, a, b) x.rbegin()+a, x.rbegin()+b+1
#define pcnt(x) __builtin_popcount(x)
#define llpcnt(x) __builtin_popcountll(x)
template<typename T>
using pq = std::priority_queue<T>;
template<typename T>
using pqg = std::priority_queue<T, std::vector<T>, std::greater<T>>;
template<typename T>
bool chmax(T &a, const T &b){
  if(a < b){ a = b; return 1; }
  else return 0;
}
template<typename T>
bool chmin(T &a, const T &b){
  if(a > b){ a = b; return 1; }
  else return 0;
}
// 多次元配列の生成
template<size_t Dimention, typename T>
class Mvector : public std::vector<Mvector<Dimention-1, T>>{
public:
  template<typename N, typename... Sizes>
  Mvector(T init, N n, Sizes... sizes) : std::vector<Mvector<Dimention-1, T> >(n, Mvector<Dimention-1, T>(init, sizes...))
  { }
};
template<typename T>
class Mvector<1, T> : public std::vector<T>{
public:
  template<typename N>
  Mvector(T init, N n) : std::vector<T>(n, init)
  { }
};


using namespace std;

#endif
0