結果

問題 No.2611 Count 01
ユーザー Sh1n3zZSh1n3zZ
提出日時 2024-01-19 22:58:12
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,823 bytes
コンパイル時間 226 ms
コンパイル使用メモリ 34,688 KB
実行使用メモリ 6,744 KB
最終ジャッジ日時 2024-01-19 22:59:37
合計ジャッジ時間 82,677 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 2 ms
6,548 KB
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cstdio>
#include <cstring>
#define MOD 998244353
#define MAXN 150010

int N, Q;
char S[MAXN];
int sum0[MAXN], sum1[MAXN], pow2[MAXN], inv2[MAXN], fac[MAXN], inv[MAXN];

int quick_pow(int a, int b) {
    int res = 1;
    while (b) {
        if (b & 1) res = 1LL * res * a % MOD;
        a = 1LL * a * a % MOD;
        b >>= 1;
    }
    return res;
}

void init() {
    pow2[0] = inv2[0] = fac[0] = inv[0] = 1;
    for (int i = 1; i <= N; ++i) {
        pow2[i] = 2LL * pow2[i - 1] % MOD;
        inv2[i] = 1LL * inv2[i - 1] * (MOD + 1) / 2 % MOD;
        fac[i] = 1LL * fac[i - 1] * i % MOD;
        inv[i] = 1LL * (MOD - MOD / i) * inv[MOD % i] % MOD;
    }
    for (int i = 2; i <= N; ++i) inv[i] = 1LL * inv[i - 1] * inv[i] % MOD;
}

int C(int n, int m) {
    if (m > n || m < 0) return 0;
    return 1LL * fac[n] * inv[m] % MOD * inv[n - m] % MOD;
}

int main() {
    scanf("%d%d%s", &N, &Q, S + 1);
    init();
    for (int i = 1; i <= N; ++i) {
        sum0[i] = sum0[i - 1] + (S[i] == '0');
        sum1[i] = sum1[i - 1] + (S[i] == '1');
    }
    while (Q--) {
        int op, x, y;
        scanf("%d%d", &op, &x);
        if (op == 1) {
            if (S[x] == '0') --sum0[x], ++sum1[x], S[x] = '1';
            else --sum1[x], ++sum0[x], S[x] = '0';
            for (int i = x + 1; i <= N; ++i) sum0[i] = sum0[i - 1], sum1[i] = sum1[i - 1];
        } else {
            scanf("%d", &y);
            int ans = 0, len = y - x + 1, cnt0 = sum0[y] - sum0[x - 1], cnt1 = sum1[y] - sum1[x - 1];
            for (int i = 0; i <= cnt0; ++i) {
                int res = 1LL * C(cnt0, i) * C(len - i, cnt1) % MOD * pow2[i] % MOD;
                if (i & 1) ans = (ans - res + MOD) % MOD;
                else ans = (ans + res) % MOD;
            }
            printf("%d\n", ans);
        }
    }
    return 0;
}
0