結果

問題 No.2541 Divide 01 String
コンテスト
ユーザー anago-pie
提出日時 2023-11-24 22:37:38
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 10 ms / 2,000 ms
コード長 674 bytes
コンパイル時間 1,881 ms
コンパイル使用メモリ 168,216 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-26 09:39:21
合計ジャッジ時間 2,606 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define rep(i, n) for(int i=0; i<n; i++)
#define debug 0
#define YES cout << "Yes" << endl;
#define NO cout << "No" << endl;
using ll = long long;
using ld = long double;
const int mod = 998244353;
const int MOD = 1000000007;
const double pi = atan2(0, -1);
#include <time.h>
#include <chrono>

int main() {
	int N;
	string S;
	cin >> N >> S;
	vector<ll> dp(N + 1, 0);
	ll c = 0;
	rep(i, N) {
		if (S[i] == '0') {
			c++;
			dp[i + 1] += dp[i];
		}
		if (S[i] == '1') {
			if (dp[i] == 0) {
				dp[i + 1] = 1;
			}
			else {
				dp[i + 1] += dp[i] * (c + 2);
			}
			c = 0;
		}
		dp[i + 1] %= mod;
	}

	cout << dp[N] << endl;
}
0