背景:
FFT的字符串假题。
题目传送门:
https://www.luogu.org/problemnew/show/P4173
题意:
定义
∗
*
∗表示一个任意字符,求一个字符串从哪里开始能在模式串完全匹配。
思路:
考虑转化为数字问题。
另
0
0
0表示
∗
*
∗,
1
1
1表示字符
a
a
a,
2
2
2表示字符
b
b
b…
考虑一一对应。
那么假设两个位置匹配,就会有
∣
a
i
−
b
i
∣
=
0
|a_i-b_i|=0
∣ai−bi∣=0或
a
i
=
0
a_i=0
ai=0或
b
i
=
0
b_i=0
bi=0,就可以表示为:
(
a
i
−
b
i
)
2
a
i
b
i
=
0
(a_i-b_i)^2a_ib_i=0
(ai−bi)2aibi=0
展开化简得: a i 2 ∗ a i b i + b i 2 ∗ a i b i − 2 a i b i ∗ a i b i = 0 a_i^2*a_ib_i+b_i^2*a_ib_i-2a_ib_i*a_ib_i=0 ai2∗aibi+bi2∗aibi−2aibi∗aibi=0
a i 3 b i + a i b i 3 − 2 a i 2 b i 2 = 0 a_i^3b_i+a_ib_i^3-2a_i^2b_i^2=0 ai3bi+aibi3−2ai2bi2=0
不妨设
b
′
b'
b′为
b
b
b反向取后的序列,则有:
a
i
3
b
n
−
i
+
a
i
b
n
−
i
3
−
2
a
i
2
b
n
−
i
2
=
0
a_i^3b_{n-i}+a_ib_{n-i}^3-2a_i^2b_{n-i}^2=0
ai3bn−i+aibn−i3−2ai2bn−i2=0
发现是三个
F
F
T
FFT
FFT卷积的形式,不妨用
F
F
T
FFT
FFT来加速即可。
代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define LL long long
using namespace std;
const double pai=acos(-1.0);
struct comp
{
double x,y;
comp(double xx=0,double yy=0):x(xx),y(yy) {}
friend comp operator+(const comp &x,const comp &y) {return comp(x.x+y.x,x.y+y.y);}
friend comp operator-(const comp &x,const comp &y) {return comp(x.x-y.x,x.y-y.y);}
friend comp operator*(const comp &a,const comp &b) {return comp(a.x*b.x-a.y*b.y,a.x*b.y+b.x*a.y);}
}a[1500000],b[1500000];
int limit=1,n,m,l=0,ma=0,ans=0;
int id[3][300010];
int r[1500000];
int mark[3][3][300010];
char s1[300010],s2[300010];
double f[1500010];
void init()
{
while(limit<=ma)
limit<<=1,l++;
for(int i=1;i<=limit;i++)
r[i]=((r[i>>1]>>1)|((i&1)<<(l-1)));
}
void FFT(comp *now,int ty)
{
for(int i=0;i<limit;i++)
if(i<r[i]) swap(now[i],now[r[i]]);
for(int mid=1;mid<limit;mid<<=1)
{
comp wn(cos(pai/mid),ty*sin(pai/mid));
for(int j=0,R=(mid<<1);j<limit;j+=R)
{
comp w(1,0);
for(int k=0;k<mid;k++,w=w*wn)
{
comp x=now[j+k],y=w*now[j+k+mid];
now[j+k]=x+y;
now[j+k+mid]=x-y;
}
}
}
}
void work(int ID)
{
for(int i=0;i<=limit;i++)
a[i]=(comp){0,0},b[i]=(comp){0,0};
for(int i=1;i<=m;i++)
a[i].x=(double)mark[ID][1][i];
for(int i=1;i<=n;i++)
b[i].x=(double)mark[ID][2][i];
FFT(a,1);
FFT(b,1);
for(int i=0;i<=limit;i++)
a[i]=a[i]*b[i];
FFT(a,-1);
if(ID!=2)
for(int i=0;i<=limit;i++)
f[i]+=a[i].x/limit;
else
for(int i=0;i<=limit;i++)
f[i]-=a[i].x/limit;
}
int main()
{
scanf("%d %d",&m,&n);
scanf("%s %s",s1+1,s2+1);
for(int i=1;i<=m;i++)
id[1][m-i+1]=s1[i]=='*'?0:s1[i]-'a'+1;
for(int i=1;i<=n;i++)
id[2][i]=s2[i]=='*'?0:s2[i]-'a'+1;
ma=n+m;
init();
for(int i=1;i<=m;i++)
mark[0][1][i]=id[1][i]*id[1][i]*id[1][i];
for(int i=1;i<=n;i++)
mark[0][2][i]=id[2][i];
for(int i=1;i<=m;i++)
mark[1][1][i]=id[1][i];
for(int i=1;i<=n;i++)
mark[1][2][i]=id[2][i]*id[2][i]*id[2][i];
for(int i=1;i<=m;i++)
mark[2][1][i]=2*id[1][i]*id[1][i];
for(int i=1;i<=n;i++)
mark[2][2][i]=id[2][i]*id[2][i];
work(0),work(1),work(2);
for(int i=m+1;i<=n+1;i++)
if(f[i]>-1e-2&&f[i]<1e-2) ans++;
printf("%d\n",ans);
for(int i=m+1;i<=n+1;i++)
if(f[i]>-1e-2&&f[i]<1e-2) printf("%d ",i-m);
}