結果

問題 No.3239 Omnibus
ユーザー shobonvip
提出日時 2025-08-16 19:11:36
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 238 ms / 10,000 ms
コード長 2,226 bytes
コンパイル時間 2,036 ms
コンパイル使用メモリ 218,076 KB
実行使用メモリ 24,652 KB
最終ジャッジ日時 2025-08-16 19:11:49
合計ジャッジ時間 9,748 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff #

/**
	author:  shobonvip
	created: 2025.08.16 18:04:39
**/

#include<bits/stdc++.h>
using namespace std;
#include<atcoder/fenwicktree>
typedef long long ll;
#define rep(i, s, n) for (int i = (int)(s); i < (int)(n); i++)
#define all(v) v.begin(), v.end()

int main(){
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	
	int n,q;
	cin>>n>>q;
	string s;
	cin>>s;
	string org=s;

	vector seg0(26*26*26,atcoder::fenwick_tree<ll>(0)); // 個数の和
	vector seg1(26*26*26,atcoder::fenwick_tree<ll>(0)); // 添字の和
	vector dat(26*26*26,vector<int>(0)); // とりうる index の集合

	auto ston = [](string s) -> int {
		return int(s[0]-'a')*26*26+int(s[1]-'a')*26+int(s[2]-'a');
	};

	rep(i,2,n){
		dat[ston(s.substr(i-2,3))].push_back(i);
	}

	vector<int> t(q),k(q),l(q),r(q);
	vector<char> c(q);
	vector<string> a(q);
	rep(i,0,q){
		cin>>t[i];
		if(t[i]==1){
			cin >> k[i] >> c[i];
			k[i]--;
			s[k[i]] = c[i];
			rep(j,max(2,k[i]),min(k[i]+3,n)){
				dat[ston(s.substr(j-2,3))].push_back(j);
			}
		}else{
			cin >> l[i] >> r[i];
			cin >> a[i];
			l[i]--;
		}
	}

	s=org;

	rep(i,0,26*26*26){
		sort(all(dat[i]));
		dat[i].erase(unique(all(dat[i])),dat[i].end());
		seg0[i] = atcoder::fenwick_tree<ll>((int)dat[i].size());
		seg1[i] = atcoder::fenwick_tree<ll>((int)dat[i].size());
	}

	auto bisect_left = [](vector<int> &dat, int i) -> int {
		return int(lower_bound(all(dat),i)-dat.begin());
	};
	
	rep(i,2,n){
		int x = ston(s.substr(i-2,3));
		int idx = bisect_left(dat[x],i);
		seg0[x].add(idx,1LL);
		seg1[x].add(idx,(ll)i);
	}

	rep(i,0,q){
		if(t[i]==1){
			rep(j,max(2,k[i]),min(k[i]+3,n)){
				int x = ston(s.substr(j-2,3));
				int idx = bisect_left(dat[x],j);
				seg0[x].add(idx,-1LL);
				seg1[x].add(idx,-(ll)j);
			}
			s[k[i]] = c[i];
			rep(j,max(2,k[i]),min(k[i]+3,n)){
				int x = ston(s.substr(j-2,3));
				int idx = bisect_left(dat[x],j);
				seg0[x].add(idx,1LL);
				seg1[x].add(idx,(ll)j);
			}
		}else{
			int x = ston(a[i]);
			if(r[i]-l[i]<=2){
				cout << 0 << '\n';
				continue;
			}
			int lb = bisect_left(dat[x],l[i]+2);
			int ub = bisect_left(dat[x],r[i]);
			ll v1 = seg1[x].sum(lb,ub);
			ll v0 = seg0[x].sum(lb,ub);
			ll ans = v1-(l[i]+1)*v0;
			cout << ans << '\n';
		}
	}
}

0