結果

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

ソースコード

diff #

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

const int MODULO=998244353;

int main(void){
    int n;
    cin >> n;
    string s;
    cin >> s;
    
    vector<long> A; //「1」の位置
    for(int i=0;i<n;i++){
        if(s[i]=='1'){
            A.push_back((long)i);
        }
    }
    
    size_t m=A.size();
    long ans=(m?1:0);
    
    for(size_t j=1;j<m;j++){
        //次の「1」までのどこで分割するか…A[j]-A[j-1]通り
        //分割しない…+1通り
        //但し、最後の「1」の後は分割しない
        ans = (ans*(A[j]-A[j-1]+1))%MODULO;
    }
    cout << ans << endl;
    return 0;
}
0