結果

問題 No.2541 Divide 01 String
コンテスト
ユーザー rikein12
提出日時 2023-11-24 23:24:04
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 7 ms / 2,000 ms
コード長 4,332 bytes
コンパイル時間 1,916 ms
コンパイル使用メモリ 193,784 KB
最終ジャッジ日時 2025-02-18 00:42:16
ジャッジサーバーID
（参考情報） 		judge3 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#line 2 "lib/base/base.hpp"

#include<bits/stdc++.h>
using namespace std;

#define rep(i,n) for(int (i)=0;(i)<(n);++(i))
#define rep3(i,m,n) for(int (i)=(m);(i)<(n);++(i))
#define endl '\n'
#define all(v) (v).begin(),(v).end()

using ll = long long;

using vi = vector<int>;
using vvi = vector<vi>;

using vll = vector<ll>;
using vvll = vector<vll>;

template <class T>
using uset = unordered_set<T>;
template <class S, class T>
using umap = unordered_map<S, T>;

using pii = pair<int, int>;
using pll = pair<ll, ll>;

template <class T>
using gprique = priority_queue<T, vector<T>, greater<T>>;
template <class T>
using lprique = priority_queue<T, vector<T>, less<T>>;

const int mod998 = 998244353;
const int mod107 = 1000000007;
#line 2 "lib/utility/io.hpp"

#define INT(...) \
  int __VA_ARGS__; \
  IN(__VA_ARGS__)
#define LL(...) \
  ll __VA_ARGS__; \
  IN(__VA_ARGS__)
#define STR(...) \
  string __VA_ARGS__; \
  IN(__VA_ARGS__)
#define CHR(...) \
  char __VA_ARGS__; \
  IN(__VA_ARGS__)
#define DBL(...) \
  long double __VA_ARGS__; \
  IN(__VA_ARGS__)

#define VEC(type, name, size) \
  vector<type> name(size);    \
  read(name)
#define VV(type, name, h, w)                     \
  vector<vector<type>> name(h, vector<type>(w)); \
  read(name)

void read(int &a) { cin >> a; }
void read(long long &a) { cin >> a; }
void read(char &a) { cin >> a; }
void read(double &a) { cin >> a; }
void read(long double &a) { cin >> a; }
void read(string &a) { cin >> a; }
template <class T, class S> void read(pair<T, S> &p) { read(p.first), read(p.second); }
template <class T> void read(vector<T> &a) {for(auto &i : a) read(i);}
template <class T> void read(T &a) { cin >> a; }
void IN() {}
template <class Head, class... Tail> void IN(Head &head, Tail &...tail) {
  read(head);
  IN(tail...);
}

template <typename T, typename U>
ostream& operator<<(ostream& os, const pair<T, U>& A) {
  os << A.first << " " << A.second;
  return os;
}

template <typename T>
ostream& operator<<(ostream& os, const vector<T>& A) {
  for (size_t i = 0; i < A.size(); i++) {
    if(i) os << " ";
    os << A[i];
  }
  return os;
}

template <typename T, typename U>
ostream& operator<<(ostream& os, const map<T, U>& A) {
  int count = 0;
  os << "{";
  for(auto [t, u] : A){
    if(count) os << ", ";
    os << t << ": " << u;
    count++;
  }
  os << "}";
  return os;
}

template <typename T>
ostream& operator<<(ostream& os, const set<T>& A) {
  for(auto it = A.begin(); it != A.end(); ++it){
    if (it != A.begin()) os << " ";
    os << (*it);
  }
  return os;
}

template <typename T>
ostream& operator<<(ostream& os, const multiset<T>& A) {
  size_t count = 0;
  T prev;
  os << "{";
  bool is_null = true, is_first = true;
  for(auto it = A.begin(); it != A.end(); ++it){
    if(it == A.begin()) {
      is_null = false;
      prev = (*it);
      count = 1;
    }
    if (prev != (*it)){
      if(!is_first) os << ", ";
      is_first = false;
      os << prev << ": " << count;
      prev = (*it);
      count = 1;
    } else {
      ++count;
    }
  }
  if(!is_null){
    if(!is_first) os << ", ";
    is_first = false;
    os << prev << ": " << count;
  }
  os << "}";
  return os;
}


void print() {
  cout << "\n";
  cout.flush();
}

template <class Head, class... Tail>
void print(Head&& head, Tail&&... tail) {
  cout << head;
  if (sizeof...(Tail)) cout << " ";
  print(forward<Tail>(tail)...);
}

void YES(bool t = 1) { print(t ? "YES" : "NO"); }
void NO(bool t = 1) { YES(!t); }
void Yes(bool t = 1) { print(t ? "Yes" : "No"); }
void No(bool t = 1) { Yes(!t); }
void yes(bool t = 1) { print(t ? "yes" : "no"); }
void no(bool t = 1) { yes(!t); }
#line 3 "main.cpp"

ll DP[200020][2];
const ll mod = 998244353;
int main(){
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    int n;
    IN(n);
    string s;
    IN(s);
    rep(i,n+1) rep(j,2)DP[i][j] = 0;
    DP[0][0] = 1;
    rep(i,n){
        if (i>0){
            if (s[i] == '1'){
                DP[i+1][1] = (DP[i][0] + 2*DP[i][1]) % mod;    
            } else {
                DP[i+1][0] = (DP[i][0] + DP[i][1]) % mod;
                DP[i+1][1] = DP[i][1];
            }
        } else {
            if (s[i] == '1'){
                DP[i+1][1] = 1;    
            } else {
                DP[i+1][0] = 1;
            }

        }
    }
    print(DP[n][1]);
    
}
0