#979579 (C++17) No.2762 Counting and Deleting

• 問題: No.2762 Counting and Deleting
• ユーザー: 👑 binap
• 提出日時: 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

ソースコード

#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;
}