yukicoder : No.2761 Substitute and Search

久々にこれやったな。
https://yukicoder.me/problems/no/2761

問題

N個のL文字の文字列S[i]が与えられる。
以下のクエリQ個に答えよ。

• Sの各k文字目がcなら、dに置き換える。
• 文字列tが与えられる。Sのうちprefixがtと一致するものを答える。

解法

BITなど使い、Sのprefixに対するrolling hashを高速に計算できるようにしよう。
またこれなら1文字の更新後のハッシュ値も高速に計算できる。

あとは前者のクエリは愚直に各文字列1文字ずつ更新したり、後者のクエリで各文字列の|t|文字prefixのhashを求めても間に合う。

template<class V, int ME> class BIT {
public:
	V bit[1<<ME];
	V operator()(int e) {if(e<0) return 0;V s=0;e++;while(e) s+=bit[e-1],e-=e&-e; return s;}
	void add(int e,V v) { e++; while(e<=1<<ME) bit[e-1]+=v,e+=e&-e;}
};
BIT<ll,12> bt[1010][3];
ll mo[3]={998244353,1000000007,1000000009};
ll p[5050][3];

int N,L,Q;
string S[1010];

void solve() {
	int i,j,k,l,r,x,y; string s;
	
	FOR(j,3) {
		p[0][j]=1;
		p[1][j]=rand();
		for(i=1;i<=3010;i++) p[i+1][j]=p[i][j]*p[1][j]%mo[j];
	}
	
	
	cin>>N>>L>>Q;
	FOR(i,N) {
		cin>>S[i];
		FOR(j,L) {
			FOR(k,3) bt[i][k].add(j,S[i][j]*p[j][k]%mo[k]);
		}
	}
	
	while(Q--) {
		cin>>i;
		if(i==1) {
			int k;
			char c,d;
			cin>>k>>c>>d;
			k--;
			FOR(i,N) if(S[i][k]==c) {
				FOR(j,3) {
					bt[i][j].add(k,-((ll)c*p[k][j]%mo[j]));
					bt[i][j].add(k,((ll)d*p[k][j]%mo[j]));
				}
				S[i][k]=d;
			}
		}
		else {
			cin>>s;
			ll ret[3]={};
			
			FOR(k,3) {
				FOR(i,s.size()) ret[k]+=(ll)s[i]*p[i][k]%mo[k];
			}
			int ok=0;
			FOR(i,N) {
				int nok=0;
				FOR(j,3) if(ret[j]==bt[i][j](s.size()-1)) nok++;
				if(nok==3) ok++;
			}
			cout<<ok<<endl;
		}
	}
}

まとめ

BITでrolling hash管理するの最近やってないので忘れてたな。