結果

問題 No.2762 Counting and Deleting
ユーザー 👑 binapbinap
提出日時 2024-05-08 00:39:51
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,881 ms / 4,000 ms
コード長 1,607 bytes
コンパイル時間 4,977 ms
コンパイル使用メモリ 287,048 KB
実行使用メモリ 42,240 KB
最終ジャッジ日時 2024-05-08 22:34:29
合計ジャッジ時間 23,321 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 1,740 ms
42,240 KB
testcase_08 AC 1,717 ms
42,088 KB
testcase_09 AC 1,740 ms
42,080 KB
testcase_10 AC 1,708 ms
42,240 KB
testcase_11 AC 1,881 ms
42,204 KB
testcase_12 AC 1,823 ms
42,212 KB
testcase_13 AC 1,860 ms
42,116 KB
testcase_14 AC 1,809 ms
42,076 KB
testcase_15 AC 1,514 ms
42,076 KB
testcase_16 AC 1,497 ms
42,240 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
#include<atcoder/all>
#define rep(i,n) for(int i=0;i<n;i++)
using namespace std;

using mint = atcoder::modint998244353;

// https://youtu.be/ylWYSurx10A?t=2352
template<typename T>
struct Matrix  : vector<vector<T>> {
	int h, w;
	Matrix(int h, int w, T val=0): vector<vector<T>>(h, vector<T>(w, val)), h(h), w(w) {}
	Matrix operator*=(const Matrix& M){
		assert(w == M.h);
		Matrix r(h, M.w);
		rep(i,h) rep(k,w) rep(j, M.w){
			r[i][j] += (*this)[i][k] * M[k][j];
		}
		swap(*this, r);
		return *this;
	}
	Matrix operator*(const Matrix& M) const {return (Matrix(*this) *= M);}
};

using S = Matrix<mint>;
S unit(3, 3);
S zero(3, 3);
S one(3, 3);


void setting(){
	unit[0] = {1, 0, 0};
	unit[1] = {0, 1, 0};
	unit[2] = {0, 0, 1};
	
	zero[0] = {1, 1, 0};
	zero[1] = {0, 1, 0};
	zero[2] = {0, 0, 1};
	
	one[0] = {1, 0, 0};
	one[1] = {1, 1, 1};
	one[2] = {0, 0, 1};
}


S op(S a, S b){
	return a * b;
}
S e(){
	return unit;
}


int main(){
	setting();
	int n, q;
	cin >> n >> q;
	string s;
	cin >> s;
	atcoder::segtree<S, op, e> seg(n);
	rep(i, n){
		if(s[i] == '0') seg.set((n - 1) - i, zero);
		if(s[i] == '1') seg.set((n - 1) - i, one);
	}
	set<int> rest;
	rep(i, n) rest.insert(i);
	rep(i, q){
		int t, l, r;
		cin >> t >> l >> r;
		l--; r--;
		if(t == 1){
			while(true){
				auto it = rest.lower_bound(l);
				if(it == rest.end()) break;
				if(*it > r) break;
				seg.set((n - 1) - *it, unit);
				rest.erase(it);
			}
		}
		if(t == 2){
			auto res = seg.prod((n - 1) - r, (n - 1) - l + 1);
			mint ans = res[0][2] + res[1][2];
			cout << ans.val() << "\n";
		}
	}
	return 0;
}
0