【bzoj4503】 两个串 FFT

$FFT$套路题（然而我看错题了）

我们考虑化一下式子。

设当前比较的两个部分为$S[i....i+|T|-1]$和$T[0....|T|-1]$。

我们对串$T$中出现问号的位置全部赋值为$0$。

我们定义一个差异度$C[i]=\sum_{j=0}^{|T|-1}T[j](S[i+j]-T[j])^2$

显然当$C[i]$为$0$时，$S[i....i+|T|-1]$和$T[0....|T|-1]$可以实现匹配。

我们把式子拆开分析，则有

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

然后我们将$T$串翻转一下，就会发现这个式子可以变成一个卷积的形式。

然后我们就可以用$FFT$去求出每一个$C[i]$，显然$T[i]^3$可以直接求。

完结撒花~

 

#include<bits/stdc++.h>
#define L long long
#define MOD 998244353
#define G 3
#define M 1<<18
using namespace std;

L pow_mod(L x,L k){
    L ans=1;
    for(;k;x=x*x%MOD,k>>=1)
    if(k&1) ans=ans*x%MOD;
    return ans;
}

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

void change(L 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 NTT(L a[],int n,int on){
    change(a,n);
    for(int h=2;h<=n;h<<=1){
        L wn=pow_mod(G,(MOD-1)/h);
        for(int j=0;j<n;j+=h){
            L w=1;
            for(int k=j;k<j+(h>>1);k++){
                L u=a[k],t=w*a[k+(h>>1)]%MOD;
                a[k]=(u+t)%MOD; 
                a[k+(h>>1)]=(u-t+MOD)%MOD;
                w=w*wn%MOD;
            }
        }
    }
    if(on==-1){
        L inv=pow_mod(n,MOD-2);
        for(int i=0;i<n;i++) a[i]=a[i]*inv%MOD;
        reverse(a+1,a+n);
    }
}

char s[M]={0},c[M]={0};
int lens,lenc,len=1;
int main(){
    scanf("%d%d",&lens,&lenc); 
    scanf("%s%s",s,c);
    lens=strlen(s); lenc=strlen(c);
    while(len<lens+lenc) len<<=1;
    reverse(c,c+lenc);
    L sumb=0;
    for(int i=0;i<lens;i++) a[i]=(s[i]-'a'+1),aa[i]=a[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],sumb+=b[i]*b[i]*b[i];
    sumb%=MOD;
    NTT(a,len,1); NTT(aa,len,1);
    NTT(b,len,1); NTT(bb,len,1);
    for(int i=0;i<len;i++) ans[i]=(aa[i]*b[i]%MOD-2*a[i]*bb[i]%MOD+MOD)%MOD;
    NTT(ans,len,-1);
    int sum=0;
    for(int i=lenc-1;i<lens;i++) 
    if((ans[i]+sumb)%MOD==0) sum++;
    cout<<sum<<endl;
    for(int i=lenc-1;i<lens;i++)
    if((ans[i]+sumb)%MOD==0) printf("%d\n",i-lenc+1);
}