CF528D Solution

题目链接

题解

差分求出\([i-k,i+k]\)区间（后文称其为 容错范围 ）中每种字母出现的次数。设bitset数组\(pos[i][j]\)表示\(s\)串第\(j\)位的容错范围内是/否（\(1/0\)）有第\(i\)种字母。对于\(t_i\)，设\(tmp=pos[t_i]>>i\)，\(tmp_j\)表示\(s_j\)为区间起点时是/否（\(1/0\)）满足存在\(t_i\)在容错范围内，将\(pos[t_i]>>i\)并在一起即为答案。

代码

#include<bits/stdc++.h>
using namespace std;
const int N=2e5+10;
char s[N],t[N];
int cha[N][5],mp[200];
bitset<N> pos[5],ans;
inline int read()
{
	int s=0,w=1; char ch=getchar();
	while(ch<'0' || ch>'9') {if(ch=='-') w=-1; ch=getchar();}
	while(ch>='0' && ch<='9') s=s*10+ch-'0',ch=getchar();
	return s*w;
}
int main()
{
	int n=read(),m=read(),k=read(),res=0;
	scanf("%s%s",s+1,t+1);
	mp['A']=1,mp['T']=2,mp['G']=3,mp['C']=4;
	for(int i=1;i<=n;i++) 
		cha[max(1,i-k)][mp[s[i]]]++,cha[min(n+1,i+k+1)][mp[s[i]]]--;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=4;j++)
		{
			if(i) cha[i][j]+=cha[i-1][j];
			if(cha[i][j]) pos[j][i]=1;
			else pos[j][i]=0;
		}
	}
	for(int i=0;i<=n;i++) ans[i]=1;
	for(int i=1;i<=m;i++) ans&=(pos[mp[t[i]]]>>i);
	for(int i=0;i<n;i++)
		if(ans[i]==1) res++;
	printf("%d",res);
	return 0;
}