【bzoj4259】 残缺的字符串 FFT

又是一道FFT套路题

思路可以参考bzoj4503，题解

我们对串S和串T中出现的*处全部赋值为0。

反正最终的差异度式子大概就是

$C[i]=\sum_{j=0}^{|T|-1}S[i+j]T[j](S[i+j]-T[j])^2$

然后和上一题一样的展开方式，将T串reverse一下做FFT再统计下即可。

然后这题卡常，FFT的长度是100W，所以用NTT会被卡常（我就T了）

然后就没了

#include<bits/stdc++.h>
#define PI acos(-1)
#define zero(x) (fabs(x)<0.5)
#define M 1<<20
using namespace std;
struct cp{
    double r,i;
    cp(){r=i=0;} cp(int x){r=x; i=0;}
    cp(double rr,double ii){r=rr;i=ii;}
    friend cp operator +(cp a,cp b){return cp(a.r+b.r,a.i+b.i);}
    friend cp operator -(cp a,cp b){return cp(a.r-b.r,a.i-b.i);}
    friend cp operator *(cp a,cp b){return cp(a.r*b.r-a.i*b.i,a.r*b.i+a.i*b.r);}
};

cp a[M]={0},aa[M]={0},aaa[M]={0};
cp b[M]={0},bb[M]={0},bbb[M]={0};
int n; cp ans[M]={0};

void change(cp a[],int n){
    for(int i=0,j=0;i<n-1;i++){
        if(i<j) swap(a[i],a[j]);
        int k=n>>1; 
        while(j>=k) j-=k,k>>=1;
        j+=k;
    }
}
void FFT(cp a[],int n,int on){
    change(a,n);
    for(int h=2;h<=n;h<<=1){
        cp wn=cp(cos(2*PI/h),on*sin(2*PI/h));
        for(int j=0;j<n;j+=h){
            cp w=cp(1,0);
            for(int k=j;k<j+(h>>1);k++){
                cp u=a[k],t=w*a[k+(h>>1)];
                a[k]=u+t; 
                a[k+(h>>1)]=u-t;
                w=w*wn;
            }
        }
    }
    if(on==-1) for(int i=0;i<n;i++) a[i].r/=n;
}

char s[M]={0},c[M]={0};
int lens,lenc,len=1;
int main(){
    scanf("%d%d",&lens,&lenc); 
    scanf("%s%s",c,s);
    lens=strlen(s); lenc=strlen(c);
    while(len<lens+lenc) len<<=1;
    reverse(c,c+lenc);
    for(int i=0;i<lens;i++) a[i]=(s[i]=='*'?0:s[i]-'a'+1),aa[i]=a[i]*a[i],aaa[i]=aa[i]*a[i];
    for(int i=0;i<lenc;i++) b[i]=(c[i]=='*'?0:c[i]-'a'+1),bb[i]=b[i]*b[i],bbb[i]=bb[i]*b[i];
    FFT(a,len,1); FFT(aa,len,1); FFT(aaa,len,1);
    FFT(b,len,1); FFT(bb,len,1); FFT(bbb,len,1);
    for(int i=0;i<len;i++) ans[i]=aaa[i]*b[i]-2*aa[i]*bb[i]+a[i]*bbb[i];
    FFT(ans,len,-1);
    int sum=0;
    for(int i=lenc-1;i<lens;i++) 
    if(zero(ans[i].r)) sum++;
    cout<<sum<<endl;
    for(int i=lenc-1;i<lens;i++)
    if(zero(ans[i].r)) printf("%d ",i-lenc+2);
}