結果

問題 No.2541 Divide 01 String
ユーザー InTheBloom
提出日時 2023-11-24 22:58:35
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 5 ms / 2,000 ms
コード長 1,813 bytes
コンパイル時間 3,899 ms
コンパイル使用メモリ 170,368 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-26 09:44:54
合計ジャッジ時間 3,821 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std;

void main () {
    int N = readln.chomp.to!int;
    string S = readln.chomp;

    solve(N, S);
}

void solve (int N, string S) {
    /*
       dp的に解く
       新しい区間を導入したとき、そこで切るか切らないかの2択
     */
    bool ok = false;
    foreach (c; S) if (c == '1') ok = true;
    if (!ok) {
        writeln(0);
        return;
    }

    const int MOD = 998244353;

    long ans = 1; // カットしないやつ

    // 右と左の0を削除
    foreach (i, c; S) if (c == '1') { S = S[i..$]; break; }
    foreach_reverse (i, c; S) if (c == '1') { S = S[0..i+1]; break; }

    int len = 0;
    int cur = 1;
    while (true) {
        if (cur == S.length) {
            break;
        }
        if (S[cur] == '1') {
            ans *= len+2;
            ans %= MOD;
            len = -1;
        }
        len++;
        cur++;
    }

    writeln(ans);
}

long modPow (long a, long x, ref const long MOD) {
    import std.exception : enforce;

    enforce(0 <= x, "x must satisfy 0 <= x");
    enforce(1 <= MOD, "MOD must satisfy 1 <= MOD");
    enforce(MOD <= int.max, "MOD must satisfy MOD*MOD <= long.max");

    // normalize
    a %= MOD; a += MOD; a %= MOD;

    long res = 1L;
    long base = a;
    while (0 < x) {
        if (0 < (x&1)) (res *= base) %= MOD;
        (base *= base) %= MOD;
        x >>= 1;
    }

    return res % MOD;
}

T modPow (T, X, Y) (T a, X x, ref const Y MOD) {
    import std.exception: enforce;

    enforce(0 <= x, "x must satisfy 0 <= x");
    enforce(1 <= MOD, "MOD must satisfy 1 <= MOD");

    // normalize
    a %= MOD; a += MOD; a %= MOD;

    T res = 1;
    T base = a;
    while (0 < x) {
        if (0 < (x&1)) (res *= base) %= MOD;
        (base *= base) %= MOD;
        x >>= 1;
    }

    return res % MOD;
}
0