BZOJ 4259 残缺的字符串

Description

很久很久以前，在你刚刚学习字符串匹配的时候，有两个仅包含小写字母的字符串A和B，其中A串长度为m，B串长度为n。可当你现在再次碰到这两个串时，这两个串已经老化了，每个串都有不同程度的残缺。

你想对这两个串重新进行匹配，其中A为模板串，那么现在问题来了，请回答，对于B的每一个位置i，从这个位置开始连续m个字符形成的子串是否可能与A串完全匹配?

Input

第一行包含两个正整数m,n(1<=m<=n<=300000)，分别表示A串和B串的长度。

第二行为一个长度为m的字符串A。

第三行为一个长度为n的字符串B。

两个串均仅由小写字母和*号组成，其中*号表示相应位置已经残缺。

Output

第一行包含一个整数k，表示B串中可以完全匹配A串的位置个数。

若k>0，则第二行输出k个正整数，从小到大依次输出每个可以匹配的开头位置（下标从1开始）。

Sample Input

3 7
a*b
aebr*ob

Sample Output

2
1 5

HINT

Source

By Claris

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

FFT~

CE到飞起……因为cstring的c写成大写了……2333

如果没有通配符'*'，那么直接用sum{(a[i]-b[j])^2}=0就可以表示字符串相匹配，但是有了通配符后就不可以了。我们把'a'~'z'用1~26表示，把'*'用0表示，那么可以用sum{(a[i]-b[j])^2*a[i]*b[j]}来表示匹配。

然后，我们拆开这个式子，（惊讶地？）发现它居然是三个式子的和，而这三个式子我们可以用三次FFT来转化，再求和，最后转化回表达式形式就是答案~

主函数实在太麻烦，我把三次FFT隔开了，这样要好看一点：）

注意：一定要清零，一定要清零，一定要清零！

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
using namespace std;
#define pi acos(-1)
#define N 1200005
#define M 300005

int n,nn,mm,l,r[N],ans[N];
double a[M],b[M];
char s[M];

struct E{
	double r,i;
	E (double u,double v) {r=u;i=v;}
	E () {}
	E operator + (E v) {return E(v.r+r,v.i+i);}
	E operator - (E v) {return E(r-v.r,i-v.i);}
	E operator * (E v) {return E(v.r*r-v.i*i,r*v.i+i*v.r);}
}c[N],f1[N],f2[N];

E operator / (E u,double v) {return E(u.r/v,u.i/v);}

void fft(E *u,int v)
{
	for(int i=0;i<n;i++) if(i<r[i]) swap(u[i],u[r[i]]);
	for(int i=1;i<n;i<<=1)
	{
		E wn(cos(pi/i),v*sin(pi/i));
		for(int j=0;j<n;j+=(i<<1))
		{
			E w(1,0);
			for(int k=0;k<i;k++,w=wn*w)
			{
				E x=u[j+k],y=w*u[i+j+k];
				u[j+k]=x+y;u[i+j+k]=x-y;
			}
		}
	}
	if(v==-1) for(int i=0;i<n;i++) u[i]=u[i]/n;
}

int main()
{
	scanf("%d%d",&nn,&mm);
	scanf("%s",s);
	for(int i=0;i<nn;i++) a[i]=s[nn-i-1]=='*' ? (double)0:(double)s[nn-i-1]-'a'+1;
	scanf("%s",s);
	for(int i=0;i<mm;i++) b[i]=s[i]=='*' ? (double)0:(double)s[i]-'a'+1;
	for(n=1;n<2*mm;n<<=1) l++;
	for(int i=0;i<n;i++) r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));
	
	for(int i=0;i<mm;i++) f1[i].r=a[i]*a[i]*a[i],f2[i].r=b[i];
	fft(f1,1);fft(f2,1);
	for(int i=0;i<n;i++) c[i]=f1[i]*f2[i];
	
	for(int i=0;i<n;i++) f1[i].r=f1[i].i=f2[i].r=f2[i].i=0;
	for(int i=0;i<mm;i++) f1[i].r=a[i]*a[i],f2[i].r=b[i]*b[i];
	fft(f1,1);fft(f2,1);
	for(int i=0;i<n;i++) c[i]=c[i]-f1[i]*f2[i]-f1[i]*f2[i];
	
	for(int i=0;i<n;i++) f1[i].r=f1[i].i=f2[i].r=f2[i].i=0;
	for(int i=0;i<mm;i++) f1[i].r=a[i],f2[i].r=b[i]*b[i]*b[i];
	fft(f1,1);fft(f2,1);
	for(int i=0;i<n;i++) c[i]=c[i]+f1[i]*f2[i];
	
	fft(c,-1);
	for(int i=nn-1;i<mm;i++) if(fabs(c[i].r)<0.5) ans[++ans[0]]=i-nn+2;
	printf("%d\n",ans[0]);
	for(int i=1;i<=ans[0];i++) printf("%d ",ans[i]);
	return 0; 
}