結果

問題 No.3500 01 String
コンテスト
ユーザー gojoxd
提出日時 2026-04-18 17:56:50
言語 C
(gcc 15.2.0)
コンパイル:
gcc-15 -O2 -DONLINE_JUDGE -o a.out _filename_ -lm
実行:
./a.out
結果
WA  
実行時間 -
コード長 994 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 387 ms
コンパイル使用メモリ 38,144 KB
実行使用メモリ 6,528 KB
最終ジャッジ日時 2026-04-18 17:56:57
合計ジャッジ時間 1,417 ms
ジャッジサーバーID
（参考情報） 		judge3_1 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 2
other AC * 3 WA * 17
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <stdio.h>

#define MAXN 500005
#define MOD 998244353

int N;
char A[MAXN];
int can1[MAXN], can0[MAXN];

int main() {
    scanf("%d", &N);
    scanf("%s", A + 1);

    // suffix: can become 1
    can1[N+1] = 0;
    for (int i = N; i >= 1; i--) {
        can1[i] = can1[i+1] || (A[i] == '1');
    }

    // prefix: can become 0
    can0[0] = 0;
    for (int i = 1; i <= N; i++) {
        can0[i] = can0[i-1] || (A[i] == '0');
    }

    long long ans = 0;

    // count monotonic strings
    for (int k = 0; k <= N; k++) {
        int ok = 1;

        if (k > 0 && !can1[k]) ok = 0;
        if (k < N && !can0[k+1]) ok = 0;

        if (ok) ans++;
    }

    // check if original is monotonic (no "01")
    int monotonic = 1;
    for (int i = 1; i < N; i++) {
        if (A[i] == '0' && A[i+1] == '1') {
            monotonic = 0;
            break;
        }
    }

    // if original is NOT monotonic, add it
    if (!monotonic) ans++;

    printf("%lld\n", ans % MOD);
    return 0;
}
0