結果

問題 No.2611 Count 01
ユーザー Carpenters-Cat
提出日時 2024-01-20 00:47:44
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,164 bytes
コンパイル時間 3,811 ms
コンパイル使用メモリ 253,648 KB
最終ジャッジ日時 2025-02-18 21:42:48
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 1
other WA * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#include <atcoder/all>
using namespace atcoder;
struct S {
	ll sum;
	ll b0;
	ll b1;
	ll a0;
	ll a1;
	ll n0;
	ll n1;
};
S op(S a, S b) {
	S c = S{0, 0, 0, 0, 0, 0, 0};
	ll al = a.n0 + a.n1, bl = b.n0 + b.n1;
	c.sum = a.sum + (a.n0 * a.n1 * bl) + b.sum + (b.n0 * b.n1 * al) + a.b0 * b.a1 + a.b1 * b.a0;
	c.b0 = a.b0 + (b.b0 + al * b.n0);
	c.b1 = a.b1 + (b.b1 + al * b.n1);
	c.a0 = b.a0 + (a.a0 + bl * a.n0);
	c.a1 = b.a1 + (a.a1 + bl * a.n1);
	c.n0 = a.n0 + b.n0;
	c.n1 = a.n1 + b.n1;
	return c;
}
S e() {
	return S{0, 0, 0, 0, 0, 0, 0};
}
S one0() {
	return S{0, 1, 0, 1, 0, 1, 0};
}
S one1() {
	return S{0, 0, 1, 0, 1, 0, 1};
}
S one(int a) {
	return (a ? one1() : one0());
}
int main () {
	int N, Q;
	cin >> N >> Q;
	string s;
	cin >> s;
	std::vector<S> A(N);
	for (int i = 0; i < N; i ++) {
		A[i] = one(s[i] - '0');
	}
	segtree<S, op, e> sg(A);
	while (Q--) {
		int t;
		cin >> t;
		if (t - 1) {
			int l, r;
			cin >> l >> r;
			l --;
			cout << sg.prod(l, r).sum << endl;
		} else {
			int i;
			cin >> i;
			i --;
			s[i] = (s[i] == '0' ? '1' : '0');
			sg.set(i, one(s[i] - '0'));
		}
	}
}
0