CF528D. Fuzzy Search

题意：DNA序列，在母串s中匹配模式串t，对于s中每个位置i，只要s[i-k]到s[i+k]中有c就认为匹配了c。求有多少个位置匹配了t

预处理\(f[i][j]\)表示位置i可以匹配字符j

分别考虑每一个字符c，对s的每个位置i求出用\(s[i,i+m-1]\)匹配t，这个字符匹配了几次

用\(a_i=[s的位置i匹配c]，\ b_i=[t_i=c]\)

那么c的匹配次数就是\(c_j=\sum\limits_{i=0}^{m-1}a_{j+i}b_i\)，位置i匹配了t当且仅当四种字符的匹配次数和等于t的长度m

~~这时候就可以考虑bitset暴力过了~~

一个常用技巧是，反转模式串(或母串)，然后就成了卷积的形式：

这样计算是没有问题的，因为b只有\([0,m-1]\)有值其他地方为0



注意处理每个字符前memset a和b！！！！！

#include <iostream>

#include <cstdio>

#include <cstring>

#include <algorithm>

#include <cmath>

using namespace std;

typedef long long ll;

const int N=(1<<20)+5, INF=1e9;

const double PI=acos(-1);

inline int read(){

    char c=getchar();int x=0,f=1;

    while(c<'0'||c>'9'){if(c=='-')f=-1;c=getchar();}

    while(c>='0'&&c<='9'){x=x*10+c-'0';c=getchar();}

    return x*f;

}

struct meow{

	double x, y;

	meow(double a=0, double b=0):x(a), y(b){}

};

meow operator +(meow a, meow b) {return meow(a.x+b.x, a.y+b.y);}

meow operator -(meow a, meow b) {return meow(a.x-b.x, a.y-b.y);}

meow operator *(meow a, meow b) {return meow(a.x*b.x-a.y*b.y, a.x*b.y+a.y*b.x);}

meow conj(meow a) {return meow(a.x, -a.y);}

typedef meow cd;

namespace FFT{

	int n, rev[N];

	void ini(int lim) {

		n=1; int k=0;

		while(n<lim) n<<=1, k++;

		for(int i=0; i<n; i++) rev[i] = (rev[i>>1]>>1) | ((i&1)<<(k-1));

	}

	void dft(cd *a, int flag) {

		for(int i=0; i<n; i++) if(i<rev[i]) swap(a[i], a[rev[i]]);

		for(int l=2; l<=n; l<<=1) {

			int m=l>>1;

			cd wn = meow(cos(2*PI/l), flag*sin(2*PI/l));

			for(cd *p=a; p!=a+n; p+=l) {

				cd w(1, 0);

				for(int k=0; k<m; k++) {

					cd t = w*p[k+m];

					p[k+m] = p[k] - t;

					p[k] = p[k] + t;

					w=w*wn;

				}

			}

		}

		if(flag==-1) for(int i=0; i<n; i++) a[i].x/=n;

	}

	void mul(cd *a, cd *b) {

		dft(a, 1); dft(b, 1);

		for(int i=0; i<n; i++) a[i]=a[i]*b[i];

		dft(a, -1);

	}

}using FFT::mul; using FFT::ini;

int n, m, k, lim, f[N][5], cnt[5], id[300];

cd a[N], b[N], c[N];

char s[N], t[N];

int ans[N];

void solve(int now) {

	memset(a, 0, sizeof(a)); memset(b, 0, sizeof(b));

	for(int i=0; i<n; i++) a[i].x = f[i][now];

	for(int i=0; i<m; i++) b[m-1-i].x = id[(int)t[i]]==now;

	mul(a, b);

	for(int i=0; i<n; i++) ans[i] += int(a[m-1+i].x+0.5);

}

int main() {

	freopen("in","r",stdin);

	n=read(); m=read(); k=read();

	lim=n+m-1; ini(lim);

	scanf("%s%s",s,t);

	id['A']=0; id['T']=1; id['C']=2; id['G']=3;

	int l=0, r=0; cnt[ id[(int)s[0]] ]++;

	for(int i=0; i<n; i++) {

		while(l<i-k) cnt[ id[(int)s[l++]] ]--;

		while(r<n-1 && r<i+k) cnt[ id[(int)s[++r]] ]++;

		for(int j=0; j<4; j++) if(cnt[j]) f[i][j]=1;

	}

	for(int i=0; i<4; i++) solve(i);

	int sum=0;

	for(int i=0; i<n; i++) if(ans[i]==m) sum++;

	printf("%d",sum);

}