結果

問題 No.2762 Counting and Deleting
コンテスト
ユーザー 👑 binap
提出日時 2024-05-08 00:39:51
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2,129 ms / 4,000 ms
コード長 1,607 bytes
コンパイル時間 4,819 ms
コンパイル使用メモリ 277,460 KB
最終ジャッジ日時 2025-02-21 11:36:47
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

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