bzoj4403 两个串

Description

兔子们在玩两个串的游戏。给定两个字符串S和T，兔子们想知道T在S中出现了几次，

分别在哪些位置出现。注意T中可能有“?”字符，这个字符可以匹配任何字符。

Input

两行两个字符串，分别代表S和T

Output

第一行一个正整数k，表示T在S中出现了几次

接下来k行正整数，分别代表T每次在S中出现的开始位置。按照从小到大的顺序输出，S下标从0开始。

两个串a,b相等（b中有通配符）当且仅当Σ(a[i]-b[i])²b[i]=0，其中a[i],b[i]为对应字符的对应编号，且通配符对应0

翻转S串并把T串用0补至与S等长后上式可化为卷积形式用fft计算

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
int N,P;
const double pi=3.14159265358979323846;
int r[];
struct cplx{
    double a,b;
    inline cplx(double r=,double i=){a=r;b=i;}
    inline cplx operator+(cplx x){return cplx(a+x.a,b+x.b);}
    inline cplx operator-(cplx x){return cplx(a-x.a,b-x.b);}
    inline cplx operator*(cplx x){return cplx(a*x.a-b*x.b,a*x.b+b*x.a);}
}a[],b[];
void dft(cplx*a,int t){
    for(int i=;i<N;i++)if(i>r[i])std::swap(a[i],a[r[i]]);
    for(int i=;i<N;i<<=){
        cplx w(cos(pi/i),t*sin(pi/i));
        for(int j=;j<N;j+=i<<){
            cplx e(),*b=a+j,*c=b+i;
            for(int k=;k<i;k++,e=e*w){
                cplx x=b[k],y=c[k]*e;
                b[k]=x+y,c[k]=x-y;
            }
        }
    }
}
char s1[],s2[];
int v1[],v2[];
int h1[],h2[];
int as[],ap=;
int l1,l2,h3=;
int main(){
    input:{
        scanf("%s%s",s1,s2);
        l1=strlen(s1)-;
        l2=strlen(s2)-;
        for(int i=;i<=l1;i++)v1[i]=s1[l1-i]-'a'+;
        for(int i=;i<=l2;i++)v2[i]=s2[i]!='?'?s2[i]-'a'+:;
        for(int i=;i<=l2;i++)h3+=v2[i]*v2[i]*v2[i];
    }
    init:{
        for(N=,P=;N<=l1+;N<<=,++P);
        N<<=,++P;
        for(int i=;i<N;i++)r[i]=r[i>>]>>|(i&)<<P;
    }
    calc:{
        for(int i=;i<=l1;i++)a[i]=cplx(v1[i]*v1[i]);
        for(int i=;i<=l1;i++)b[i]=cplx(v2[i]);
        for(int i=l1+;i<N;i++)a[i]=b[i]=cplx();
        dft(a,),dft(b,);
        for(int i=;i<N;i++)a[i]=a[i]*b[i];
        dft(a,-);
        for(int i=;i<=l1;i++)h1[i]=(int)(a[i].a/N+0.5);
 
        for(int i=;i<=l1;i++)a[i]=cplx(v1[i]);
        for(int i=;i<=l1;i++)b[i]=cplx(v2[i]*v2[i]);
        for(int i=l1+;i<N;i++)a[i]=b[i]=cplx(,);
        dft(a,),dft(b,);
        for(int i=;i<N;i++)a[i]=a[i]*b[i];
        dft(a,-);
        for(int i=;i<=l1;i++)h2[i]=(int)(a[i].a/N+0.5);
    }
    output:{
        for(int i=l1;i>=l2;i--)if(h1[i]-*h2[i]+h3==)as[ap++]=l1-i;
        printf("%d\n",ap);
        for(int i=;i<ap;i++)printf("%d\n",as[i]);
    }
    return ;
}