#935989 (C++17) No.2541 Divide 01 String

• 問題: No.2541 Divide 01 String
• ユーザー: *yudai25n
• 提出日時: 2023-12-03 22:16:47
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 9 ms / 2,000 ms
• コード長: 1,683 bytes
• コンパイル時間: 4,567 ms
• コンパイル使用メモリ: 261,752 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-09-26 22:19:13
• 合計ジャッジ時間: 4,923 ms
• ジャッジサーバーID （参考情報）: judge4 / judge5

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	3 ms; 5,248 KB
testcase_01	AC	3 ms; 5,376 KB
testcase_02	AC	3 ms; 5,376 KB
testcase_03	AC	4 ms; 5,376 KB
testcase_04	AC	6 ms; 5,376 KB
testcase_05	AC	9 ms; 5,376 KB
testcase_06	AC	6 ms; 5,376 KB
testcase_07	AC	8 ms; 5,376 KB
testcase_08	AC	9 ms; 5,376 KB
testcase_09	AC	9 ms; 5,376 KB
testcase_10	AC	7 ms; 5,376 KB
testcase_11	AC	7 ms; 5,376 KB
testcase_12	AC	8 ms; 5,376 KB
testcase_13	AC	3 ms; 5,376 KB
testcase_14	AC	3 ms; 5,376 KB
testcase_15	AC	3 ms; 5,376 KB
testcase_16	AC	3 ms; 5,376 KB
testcase_17	AC	3 ms; 5,376 KB
testcase_18	AC	6 ms; 5,376 KB
testcase_19	AC	6 ms; 5,376 KB
testcase_20	AC	6 ms; 5,376 KB
testcase_21	AC	5 ms; 5,376 KB
testcase_22	AC	5 ms; 5,376 KB

ソースコード

#include <bits/stdc++.h>
#include <atcoder/all>
#define OVERLOAD_REP(_1, _2, _3, name, ...) name
#define REP1(i,n) for (auto i = std::decay_t<decltype(n)>{}; (i) != (n); ++(i))
#define REP2(i,l,r) for (auto i = (l); (i) != (r); ++(i))
#define rep(...) OVERLOAD_REP(__VA_ARGS__, REP2, REP1)(__VA_ARGS__)
#define all(a) (a).begin(),(a).end()
#define rall(a) (a).rbegin(),(a).rend()
using namespace std;
using namespace atcoder;
using ll = long long;
using ull = unsigned long long;
using P = pair<int,int>;
using LP = pair<long long,long long>;
using SP = pair<string,string>;
using vi = vector<int>;
using vl = vector<long long>;
using vs = vector<string>;
using vc = vector<char>;
using mint1 = modint1000000007;
using mint9 = modint998244353;
const int INF = 2e9;
const long long LINF = 4e18;
const int di[] = {0,1,0,-1,-1,1,-1,1};
const int dj[] = {1,0,-1,0,1,1,-1,-1};
template<class T> inline bool chmin(T& a, T b){if(a>b){a=b;return true;}return false;}
template<class T> inline bool chmax(T& a, T b){if(a<b){a=b;return true;}return false;}
void YesNo (bool ok , const string &s , const string &t) {if(ok) cout << s << endl; else cout << t << endl;} 


//-----def end-----//

mint9 dp[200005][2];

int main () {
    int n;
    string s;
    cin >> n >> s;
    dp[0][0] = 1;
    rep(i , n) rep(j , 2) {
        if(j == 0) {
            if(s[i] == '1') dp[i+1][1] += dp[i][j];
            else dp[i+1][0] += dp[i][j];
        } else {
            //繋げる
            dp[i+1][1] += dp[i][j];
            //分割する
            if(s[i] == '1') dp[i+1][1] += dp[i][j];
            else dp[i+1][0] += dp[i][j];
        }
    }
    cout << dp[n][1].val() << endl;
	return 0;
}